From 51540cefc448684d5086297d23e9a1805da4d405 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 1 Jul 2024 11:42:36 +0300 Subject: [PATCH] Update. --- main.py | 3 +++ quizz_machine.py | 14 ++++++++------ 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/main.py b/main.py index 11eb8fd..fd8ab41 100755 --- 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 --- 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, ) -- 2.20.1