From 574cee909dc112dbbcad960fbf1fa191721ddd1b Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 24 Sep 2024 12:41:42 +0200 Subject: [PATCH] Update. --- main.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 20ffec7..a70c758 100755 --- 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() -- 2.39.5