Update.
authorFrançois Fleuret <francois@fleuret.org>
Sat, 22 Jun 2024 21:58:45 +0000 (23:58 +0200)
committerFrançois Fleuret <francois@fleuret.org>
Sat, 22 Jun 2024 21:58:45 +0000 (23:58 +0200)
main.py

diff --git a/main.py b/main.py
index b6f2783..45fa68c 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -404,7 +404,7 @@ if args.check:
     nb_new_quizzes_for_test = 10
 
 for n_epoch in range(args.nb_epochs):
-    a = [(model.id, model.main_test_accuracy.item()) for model in models]
+    a = [(model.id, model.main_test_accuracy) for model in models]
     a.sort(key=lambda p: p[0])
     log_string(f"current accuracies {a}")