Update train.py

error030 · Oct 12, 2022 · 63a3063 · 63a3063
1 parent 7992e90
commit 63a3063
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main/train.py b/main/train.py
@@ -31,7 +31,7 @@ def parse_args():
         "--cfg",
         help="decide which cfg to use",
         required=False,
-        default="configs/cifar10.yaml",
+        default="../main/configs/cifar10.yaml",
         type=str,
     )
     parser.add_argument(