Update.

diff --git a/main.py b/main.py
index 902c376..c7131c3 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -911,7 +911,7 @@ def save_models(models, suffix=""):
             os.path.join(args.result_dir, filename),
         )
 
-    log_string(f"wrote ae_*{prefix}.pth")
+    log_string(f"wrote ae_*{suffix}.pth")
 
 
 ######################################################################