Update.

[culture.git] / main.py

diff --git a/main.py b/main.py
index ba5f04b..3004f9c 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -437,7 +437,7 @@ def create_c_quizzes(
             if c_quizzes.size(0) > 0:
                 logproba = c_quizzes.new(c_quizzes.size(0), len(models))
                 for q, l in zip(
-                    c_quizzes.split(args.batch_size), logits.split(args.batch_size)
+                    c_quizzes.split(args.batch_size), logproba.split(args.batch_size)
                 ):
                     for model in models:
                         l[model.id] = F.cross_entropy(model(q))