Update.

[culture.git] / main.py

diff --git a/main.py b/main.py
index 585cbdf..597ec32 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -57,7 +57,7 @@ parser.add_argument("--nb_train_samples", type=int, default=None)
 
 parser.add_argument("--nb_test_samples", type=int, default=None)
 
-parser.add_argument("--learning_rate", type=float, default=1e-3)
+parser.add_argument("--learning_rate", type=float, default=5e-4)
 
 ########################################
 
@@ -484,6 +484,8 @@ def create_c_quizzes(
             quizzes_and_nb_correct_records, criteria=lambda nb_correct: nb_correct == n
         )[:72]
 
+        quiz_machine.reverse_random_half_in_place(q)
+
         if q.size(0) > 0:
             quiz_machine.save_quizzes(
                 args.result_dir, f"culture_c_quiz_{n_epoch:04d}_N{n}{s}", q