fix: set skip setting from argument

DeNA · Jan 26, 2022 · ddcb6f7 · ddcb6f7
1 parent 154d283
commit ddcb6f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/aux_swa.py b/scripts/aux_swa.py
@@ -24,7 +24,7 @@
 ed, length = int(sys.argv[1]), int(sys.argv[2])
 step = 1
 if len(sys.argv) >= 4:
-    step
+    step = int(sys.argv[3])
 
 model_ids = [str(i) + '.pth' for i in range(ed - length + 1, ed + 1, step)]