X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=80f2733c7a3e97d8c8f20a8df8d7a7ad674897bb;hb=994d2408781ebaed6da16b10b2b3ebedeff82756;hp=abed321652da8a50a8336958c766b9767d0ba5a8;hpb=e6e0350e49284c1b5feb8b1b3c6929ee03a857dc;p=picoclvr.git

diff --git a/main.py b/main.py
index abed321..80f2733 100755
--- a/main.py
+++ b/main.py
@@ -127,6 +127,10 @@ parser.add_argument("--expr_nb_variables", type=int, default=5)
 
 parser.add_argument("--expr_sequence_length", type=int, default=40)
 
+parser.add_argument("--expr_operand_max", type=int, default=9)
+
+parser.add_argument("--expr_result_max", type=int, default=99)
+
 parser.add_argument("--expr_input_file", type=str, default=None)
 
 ######################################################################
@@ -307,6 +311,8 @@ elif args.task == "expr":
         nb_test_samples=args.nb_test_samples,
         nb_variables=args.expr_nb_variables,
         sequence_length=args.expr_sequence_length,
+        operand_max=args.expr_operand_max,
+        result_max=args.expr_result_max,
         batch_size=args.batch_size,
         device=device,
     )