Update.

diff --git a/main.py b/main.py
index a021a71..3b5a7c6 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -415,6 +415,8 @@ if args.check:
 desired_average_logits = None
 
 for n_epoch in range(args.nb_epochs):
+    log_string(f"--- epoch {n_epoch+1} ----------------------------------------")
+
     a = [(model.id, float(model.main_test_accuracy)) for model in models]
     a.sort(key=lambda p: p[0])
     log_string(f"current accuracies {a}")