From 9df1bd18f930f6b4a30b94fed6de684d5ceae3b7 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 22 Jun 2024 23:58:45 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index b6f2783..45fa68c 100755 --- 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}") -- 2.20.1