Update.
authorFrançois Fleuret <francois@fleuret.org>
Sun, 21 Jul 2024 03:29:08 +0000 (05:29 +0200)
committerFrançois Fleuret <francois@fleuret.org>
Sun, 21 Jul 2024 03:29:08 +0000 (05:29 +0200)
main.py

diff --git a/main.py b/main.py
index 653f5f5..5ce9731 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -533,7 +533,7 @@ def record_new_c_quizzes(models, quiz_machine, nb_for_train=1000, nb_for_test=10
             args.result_dir, prefix, vq, show_part_to_predict=False
         )
 
-    vq = too_simple_quizzes
+    vq = too_simple_quizzes[torch.randperm(too_simple_quizzes.size(0))[:128]]
 
     if vq.size(0) > 0:
         prefix = f"culture_c_quiz_{n_epoch:04d}_too_simple"