From: François Fleuret <francois@fleuret.org>
Date: Wed, 26 Jun 2024 15:18:06 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=07458603bfb24d5a12b530839e52e42fe8b0e6b8;p=culture.git

Update.
---

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))[