From 07458603bfb24d5a12b530839e52e42fe8b0e6b8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 26 Jun 2024 17:18:06 +0200 Subject: [PATCH] Update. --- quizz_machine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quizz_machine.py b/quizz_machine.py index f799bf1..8ee0226 100755 --- a/quizz_machine.py +++ b/quizz_machine.py @@ -113,7 +113,7 @@ class QuizzMachine: if len(c_quizzes) > 0: c_quizzes = torch.cat(c_quizzes, dim=0) if c_quizzes.size(0) > w_quizzes.size(0) // 2: - i = torch.randperm(w_quizzes.size(0))[: w_quizzes.size(0) // 2] + i = torch.randperm(c_quizzes.size(0))[: w_quizzes.size(0) // 2] c_quizzes = c_quizzes[i] i = torch.randperm(w_quizzes.size(0))[ -- 2.20.1