From: François Fleuret <francois@fleuret.org>
Date: Sat, 21 Sep 2024 14:59:01 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=6c6c62671ded6262be867ca1ac43c7fefd812b90;p=culture.git

Update.
---

diff --git a/main.py b/main.py
index 0056c76..cd9ec20 100755
--- a/main.py
+++ b/main.py
@@ -282,14 +282,18 @@ def generate_quiz_set(nb_samples, c_quizzes, c_quiz_multiplier=1):
             body = c_quizzes.repeat(n, 1)
             if n < c_quiz_multiplier:
                 tail = c_quizzes[
-                    torch.randperm(c_quizzes.size(0))[: nb_samples // 2 - body.size(0)]
+                    torch.randperm(c_quizzes.size(0), device=c_quizzes.device)[
+                        : nb_samples // 2 - body.size(0)
+                    ]
                 ]
                 c_quizzes = torch.cat([body, tail], dim=0)
             else:
                 c_quizzes = body
 
         if c_quizzes.size(0) > nb_samples // 2:
-            i = torch.randperm(c_quizzes.size(0))[: nb_samples // 2]
+            i = torch.randperm(c_quizzes.size(0), device=c_quizzes.device)[
+                : nb_samples // 2
+            ]
             c_quizzes = c_quizzes[i]
 
         w_quizzes = problem.generate_w_quizzes(nb_samples - c_quizzes.size(0))