Update.

diff --git a/main.py b/main.py
index 20ffec7..a70c758 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -903,6 +903,8 @@ log_string(f"vocabulary_size {vocabulary_size}")
 
 ######################################################################
 
+train_c_quizzes, test_c_quizzes = None, None
+
 models = []
 
 for i in range(args.nb_models):
@@ -948,12 +950,11 @@ if args.resume:
 nb_parameters = sum(p.numel() for p in models[0].parameters())
 log_string(f"nb_parameters {nb_parameters} ({int(nb_parameters/1e6)}M)")
 
-
 ######################################################################
 
-train_c_quizzes, test_c_quizzes = None, None
-
-######################################################################
+# for model in models:
+# inject_plasticity(model, args.proba_plasticity)
+# model.test_accuracy = 0
 
 for n_epoch in range(current_epoch, args.nb_epochs):
     start_time = time.perf_counter()