Update.

[culture.git] / quiz_machine.py

diff --git a/quiz_machine.py b/quiz_machine.py
index c1477c9..0ae68d0 100755 (executable)
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -429,7 +429,7 @@ class QuizMachine:
                     F.cross_entropy(output.transpose(1, 2), input, reduction="none")
                     * ar_mask
                 )
-                l[:, model.id] = ce.sum(dim=-1)
+                l[:, model.id] = -ce.sum(dim=-1)
 
         return logproba