Fix training not working

blaisewf · Apr 20, 2024 · 361a1c1 · 361a1c1
1 parent 888abf6
commit 361a1c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -1177,12 +1177,12 @@ def main():
                 str(args.batch_size),
                 str(args.gpu),
                 str(args.pitch_guidance),
+                str(args.overtraining_detector),
+                str(args.overtraining_threshold),
                 str(args.pretrained),
                 str(args.custom_pretrained),
                 str(args.g_pretrained_path),
                 str(args.d_pretrained_path),
-                str(args.overtraining_detector),
-                str(args.overtraining_threshold),
             )
         elif args.mode == "index":
             run_index_script(