From 10884f246fb97c69cdaa74052b2aa9e16b8be134 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 21 Aug 2024 23:18:39 +0200 Subject: [PATCH] Update. --- main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 66b7ffd..148a917 100755 --- a/main.py +++ b/main.py @@ -838,6 +838,7 @@ if args.resume: current_epoch = state["current_epoch"] total_time_generating_c_quizzes = state["total_time_generating_c_quizzes"] total_time_training_models = state["total_time_training_models"] + common_c_quiz_bags = state["common_c_quiz_bags"] except FileNotFoundError: log_string(f"cannot find {filename}") pass @@ -1060,8 +1061,9 @@ if args.test == "entropy": for n_epoch in range(current_epoch, args.nb_epochs): state = { "current_epoch": n_epoch, - "total_time_training_models": total_time_training_models, "total_time_generating_c_quizzes": total_time_generating_c_quizzes, + "total_time_training_models": total_time_training_models, + "common_c_quiz_bags": common_c_quiz_bags, } filename = "state.pth" torch.save(state, os.path.join(args.result_dir, filename)) -- 2.39.5