Update.

diff --git a/main.py b/main.py
index 3b5a7c6..2afe61b 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -415,7 +415,7 @@ if args.check:
 desired_average_logits = None
 
 for n_epoch in range(args.nb_epochs):
 desired_average_logits = None
 
 for n_epoch in range(args.nb_epochs):

-    log_string(f"--- epoch {n_epoch+1} ----------------------------------------")
+    log_string(f"--- epoch {n_epoch} ----------------------------------------")

 
     a = [(model.id, float(model.main_test_accuracy)) for model in models]
     a.sort(key=lambda p: p[0])
 
     a = [(model.id, float(model.main_test_accuracy)) for model in models]
     a.sort(key=lambda p: p[0])