Update.

[culture.git] / main.py

diff --git a/main.py b/main.py
index 585cbdf..02e1a8d 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)
 
 ########################################
 
@@ -126,6 +126,7 @@ if args.result_dir is None:
 
 if args.dirty_debug:
     args.accuracy_to_make_c_quizzes = 0.0
+    args.nb_gpts = 2
     nb_new_c_quizzes_for_train = 100
     nb_new_c_quizzes_for_test = 10
 
@@ -484,6 +485,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