X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=world.py;h=43126d5d63466e948317839e727a410e2b267c62;hb=6917d3d52a4b473d31121a471ab98fa114bdb1a6;hp=97c7b1d5dd3d85ebe9889caad1691e5df302e160;hpb=0ab695df8f6a2a0cc70a424e57943a0d5606903b;p=culture.git

diff --git a/world.py b/world.py
index 97c7b1d..43126d5 100755
--- a/world.py
+++ b/world.py
@@ -34,7 +34,7 @@ def generate(
     nb,
     height,
     width,
-    max_nb_obj=colors.size(0) - 2,
+    max_nb_obj=2,
     nb_iterations=2,
 ):
     f_start = torch.zeros(nb, height, width, dtype=torch.int64)