From: François Fleuret <francois@fleuret.org>
Date: Mon, 12 Aug 2024 22:07:15 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=ce7b4d55087703e6f67da2c26c9bed22f58113eb;p=culture.git

Update.
---

diff --git a/main.py b/main.py
index 0a79323..bd46948 100755
--- a/main.py
+++ b/main.py
@@ -540,8 +540,7 @@ def save_additional_results(model, models, science_w_quizzes):
         for model in models
     ]
 
-    seq_logprobas = torch.cat([x[None, :] for x in l])
-
+    seq_logprobas = torch.cat([x[:, None] for x in l], dim=1)
     probas = seq_logprobas.exp()
 
     comments = []