reverted depthwise flag to default False

LostOxygen · Jun 16, 2023 · be5d0f6 · be5d0f6
1 parent 39d7c86
commit be5d0f6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/main.py b/main.py
@@ -159,12 +159,10 @@ def main(gpu: int, batch_size: int, epochs: int, model_type: str,
     parser.add_argument("--model_type", "-m", help="specifies the model architecture",
                         type=str, default="vgg11")
     parser.add_argument("--depthwise", "-d", help="enables depthwise conv",
-                        action="store_true", default=True)
+                        action="store_true", default=False)
     parser.add_argument("--eval_only", "-ev", help="evaluates the model without training",
                         action="store_true", default=False)
     parser.add_argument("--normalize", "-no", help="enables normalization",
                         action="store_true", default=True)
     args = parser.parse_args()
     main(**vars(args))
-
-    #python main.py --gpu 2 --batch_size 8 --learning_rate 0.0001 --epochs 100 --model_type vgg19 --eval_only