X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=inline;f=main.py;h=597ec32f71f5232ddbbc7d26514ce1d238112816;hb=ceddc8cc3adbb045fdef1ccb0b3df2b8fed9eb4c;hp=585cbdfc80116fe8210d3eaaf9ef71a1cb87aa15;hpb=da54f88590586ec4949f8bf99bca16bc3b287097;p=culture.git diff --git a/main.py b/main.py index 585cbdf..597ec32 100755 --- 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