Update main.py

main.py

@@ -11,16 +11,24 @@ def run_transfer_learning(dataset, epochs, batch_size, imgsz, patience, cache, p
         "--imgsz", str(imgsz),
         "--patience", str(patience),
         "--cache", cache,
-        "--pretrained", str(pretrained),
-        "--cos_lr", str(cos_lr),
-        "--profile", str(profile),
-        "--plots", str(plots),
-        "--resume", str(resume),
-        "--augment", str(augment),
         "--model", model,
         "--run", run
     ]
 
+    # Add boolean flags only if they are True
+    if pretrained:
+        command.append("--pretrained")
+    if cos_lr:
+        command.append("--cos_lr")
+    if profile:
+        command.append("--profile")
+    if plots:
+        command.append("--plots")
+    if resume:
+        command.append("--resume")
+    if augment:
+        command.append("--augment")
+
     # Use subprocess to run the script with the arguments
     subprocess.run(command, check=True)