Update.

diff --git a/main.py b/main.py
index 66b7ffd..148a917 100755 (executable)
--- 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))