X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=e2b705dc8efa083484d8bd01f7fc1ad4cd7cf011;hb=e244104e7b697b79e8500b5d648ec161c4ed9a63;hp=003028a819ccc5e8882435dbb817ca59aa128ba0;hpb=5aee50805cfad1dd49bbf30b30fe65b05e03de78;p=picoclvr.git

diff --git a/main.py b/main.py
index 003028a..e2b705d 100755
--- a/main.py
+++ b/main.py
@@ -125,7 +125,9 @@ 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)
 
 ######################################################################
 
@@ -366,6 +368,20 @@ else:
 
 ######################################################################
 
+if args.task == "expr" and args.expr_input_file is not None:
+    task.produce_results(
+        nb_epochs_finished,
+        model,
+        args.result_dir,
+        log_string,
+        args.deterministic_synthesis,
+        args.expr_input_file,
+    )
+
+    exit(0)
+
+######################################################################
+
 nb_epochs = args.nb_epochs if args.nb_epochs > 0 else nb_epochs_default
 
 # Compute the entropy of the training tokens