From f3fcb3c243084d07be4a6d48178889086215e86c Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 17 Jul 2024 06:49:18 +0200 Subject: [PATCH] Update. --- quiz_machine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quiz_machine.py b/quiz_machine.py index 14e8739..bbd1b7b 100755 --- a/quiz_machine.py +++ b/quiz_machine.py @@ -341,7 +341,7 @@ class QuizMachine: ) < w_quizzes.size(0) i = torch.randperm(quizzes.size(0), device=quizzes.device) - return quizzes[i], type_w[i] + return quizzes[i], from_w[i] else: return w_quizzes, torch.full( -- 2.20.1