X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=world.py;h=c33a584320f236f9f7fa51fe7f8faec4c5de278c;hb=21be996a2e3302c952fc75354f41e359266b2941;hp=61a07e9855bca6439c8c1e4b553a5b3ffced51ea;hpb=910434daa74525b29d7ba117312fc67789d1ad84;p=picoclvr.git

diff --git a/world.py b/world.py
index 61a07e9..c33a584 100755
--- a/world.py
+++ b/world.py
@@ -388,9 +388,7 @@ def create_data_and_processors(nb_train_samples, nb_test_samples, nb_epochs=10):
 if __name__ == "__main__":
     train_input, test_input, frame2seq, seq2frame = create_data_and_processors(
         # 10000, 1000,
-        100,
-        100,
-        nb_epochs=2,
+        100, 100, nb_epochs=2,
     )
 
     input = test_input[:64]