typo in save name

atong01 · Dec 13, 2023 · fc1a870 · fc1a870
1 parent 4e141e8
commit fc1a870
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/cifar10/train_cifar10.py b/examples/cifar10/train_cifar10.py
@@ -159,7 +159,7 @@ def train(argv):
                         "optim": optim.state_dict(),
                         "step": step,
                     },
-                    savedir + f"para_cifar10_weights_step_{step}.pt",
+                    savedir + f"cifar10_weights_step_{step}.pt",
                 )