Update.

diff --git a/main.py b/main.py
index b2b0492..20ffec7 100755 (executable)
--- 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")