Update.

diff --git a/main.py b/main.py
index d2ac744..c655084 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -663,7 +663,10 @@ for n_epoch in range(args.nb_epochs):
     log_string(
         f"cache_w_quizzes contains {quiz_machine.problem.nb_cached_quizzes()} quizzes"
     )
-    quiz_machine.renew_w_quizzes(model, args.nb_train_samples // args.nb_gpts)
+
+    # Renew entirely the train set
+
+    quiz_machine.renew_w_quizzes(model, args.nb_train_samples)
 
     ##################################################
     # If all the models are good enough, generate new quizzes and