Update.

diff --git a/main.py b/main.py
index 11eb8fd..fd8ab41 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -79,6 +79,8 @@ parser.add_argument("--dropout", type=float, default=0.1)
 
 parser.add_argument("--deterministic_synthesis", action="store_true", default=False)
 
+parser.add_argument("--reverse_cleanup", action="store_true", default=False)
+
 parser.add_argument("--problem", type=str, default="sky")
 
 parser.add_argument("--nb_gpts", type=int, default=5)
@@ -418,6 +420,7 @@ def create_c_quizzes(
             nb_models_for_generation=args.nb_models_for_generation,
             models=models,
             mode=args.generation_mode,
+            reverse_cleanup=args.reverse_cleanup,
             min_ave_seq_logproba=min_ave_seq_logproba,
             n_epoch=n_epoch,
             result_dir=args.result_dir,

diff --git a/quizz_machine.py b/quizz_machine.py
index eae256b..806dde7 100755 (executable)
--- a/quizz_machine.py
+++ b/quizz_machine.py
@@ -471,16 +471,18 @@ class QuizzMachine:
         models,
         mode,
         min_ave_seq_logproba,
+        reverse_cleanup,
         n_epoch,
         result_dir,
     ):
         model_for_generation = Gang(models, nb_models_for_generation, mode)
         models_for_validation = models
         return self.create_c_quizzes(
-            nb,
-            model_for_generation,
-            models_for_validation,
-            min_ave_seq_logproba,
-            n_epoch,
-            result_dir,
+            nb=nb,
+            model_for_generation=model_for_generation,
+            models_for_validation=models_for_validation,
+            min_ave_seq_logproba=min_ave_seq_logproba,
+            reverse_cleanup=reverse_cleanup,
+            n_epoch=n_epoch,
+            result_dir=result_dir,
         )