X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=main.py;h=abed321652da8a50a8336958c766b9767d0ba5a8;hb=e6e0350e49284c1b5feb8b1b3c6929ee03a857dc;hp=56b7e1c74e40d1cbfba6d612439fb6051f70f050;hpb=2175a87ad4304a97c63ac9cca6224d0c0b74c64e;p=picoclvr.git

diff --git a/main.py b/main.py
index 56b7e1c..abed321 100755
--- a/main.py
+++ b/main.py
@@ -125,7 +125,7 @@ parser.add_argument("--stack_fraction_values_for_train", type=float, default=0.7
 
 parser.add_argument("--expr_nb_variables", type=int, default=5)
 
-parser.add_argument("--expr_sequence_length", type=int, default=30)
+parser.add_argument("--expr_sequence_length", type=int, default=40)
 
 parser.add_argument("--expr_input_file", type=str, default=None)
 
@@ -172,9 +172,9 @@ default_args = {
         "nb_test_samples": 1000,
     },
     "expr": {
-        "nb_epochs": 50,
+        "nb_epochs": 40,
         "batch_size": 25,
-        "nb_train_samples": 250000,
+        "nb_train_samples": 1000000,
         "nb_test_samples": 10000,
     },
 }