From 30d56d81514f5c7336fc437e14e0154ab09f6272 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 21 Jul 2024 05:29:08 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 653f5f5..5ce9731 100755 --- 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" -- 2.20.1