X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=quizz_machine.py;h=2cc6cfd59d02fcb417834a34455a1f42b7dfcbd5;hb=a2346746c9b417eaf97aad87ed31dea92c3bb887;hp=d63855c486673eb86211bad50781397e61a9ffd0;hpb=e2c3b8046c3fddef8aacb74cf5f848d42044897e;p=culture.git

diff --git a/quizz_machine.py b/quizz_machine.py
index d63855c..2cc6cfd 100755
--- a/quizz_machine.py
+++ b/quizz_machine.py
@@ -98,7 +98,7 @@ class QuizzMachine:
 
         if result_dir is not None:
             self.problem.save_quizzes(
-                self.train_w_quizzes[:72], result_dir, f"culture_w_quizzes", logger
+                self.train_w_quizzes[:72], result_dir, "culture_w_quizzes"
             )
 
     def batches(self, split="train", desc=None):
@@ -206,10 +206,7 @@ class QuizzMachine:
         )
 
         self.problem.save_quizzes(
-            result[:72],
-            result_dir,
-            f"culture_prediction_{n_epoch:04d}_{model.id:02d}",
-            logger,
+            result[:72], result_dir, f"culture_prediction_{n_epoch:04d}_{model.id:02d}"
         )
 
         return main_test_accuracy