From 9eb886ce64c2d82584a5a6688ee770eda45866cd Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 24 Sep 2024 12:30:26 +0200 Subject: [PATCH] Update. --- main.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index b2b0492..20ffec7 100755 --- a/main.py +++ b/main.py @@ -90,7 +90,7 @@ parser.add_argument("--gpus", type=str, default="all") parser.add_argument("--nb_models", type=int, default=5) -parser.add_argument("--proba_plasticity", type=float, default=0.0) +parser.add_argument("--proba_plasticity", type=float, default=0.25) parser.add_argument("--diffusion_nb_iterations", type=int, default=25) @@ -606,13 +606,13 @@ def one_complete_epoch( one_epoch(model, n_epoch, test_c_quizzes, train=False, local_device=local_device) # Compute the test accuracy *WITHOUT C QUIZZES* - warnings.warn("test_accuracy does not consider c quizzes", RuntimeWarning) + # warnings.warn("test_accuracy does not consider c quizzes", RuntimeWarning) quizzes = generate_quiz_set( args.nb_test_samples, - c_quizzes=None, - c_quiz_multiplier=args.c_quiz_multiplier - # args.nb_test_samples, test_c_quizzes, args.c_quiz_multiplier + # c_quizzes=None, + c_quizzes=test_c_quizzes, + c_quiz_multiplier=args.c_quiz_multiplier, ) imt_set = samples_for_prediction_imt(quizzes.to(local_device)) result = ae_predict(model, imt_set, local_device=local_device).to("cpu") -- 2.39.5