Update experiment/transfer_learning_train&test.py

experiment/transfer_learning_train&test.py

@@ -25,12 +25,12 @@ def parse_args():
     parser.add_argument("--imgsz", type=int, default=640, help="Image size")
     parser.add_argument("--patience", type=int, default=30, help="Patience for early stopping")
     parser.add_argument("--cache", type=str, default="ram", help="Cache option")
-    parser.add_argument("--pretrained", type=bool, default=False, help="Use pretrained weights")
-    parser.add_argument("--cos_lr", type=bool, default=False, help="Use cosine learning rate")
-    parser.add_argument("--profile", type=bool, default=False, help="Profile the training")
-    parser.add_argument("--plots", type=bool, default=True, help="Generate plots")
-    parser.add_argument("--resume", type=bool, default=False, help="Resume run")
-    parser.add_argument("--augment", type=bool, default=False, help="Apply augmentation techniques during training")
+    parser.add_argument("--pretrained", action="store_true", help="Use pretrained weights")
+    parser.add_argument("--cos_lr", action="store_true", help="Use cosine learning rate")
+    parser.add_argument("--profile", action="store_true", help="Profile the training")
+    parser.add_argument("--plots", action="store_true", help="Generate plots")
+    parser.add_argument("--resume", action="store_true", help="Resume run")
+    parser.add_argument("--augment", action="store_true", help="Apply augmentation techniques during training")
     parser.add_argument("--model", type=str, required=True, help="Model name")
     parser.add_argument("--run", type=str, required=True, help="Run mode")