Update inference_2.py

inference_2.py

@@ -56,11 +56,12 @@ def get_args(parser):
     parser.add_argument("--n_classes", type=int, default=2)
     parser.add_argument("--annealing_epoch", type=int, default=10)
     parser.add_argument("--device", type=str, default='cpu')
-    parser.add_argument("--pretrained_image_encoder", type=bool, default = False)
-    parser.add_argument("--freeze_image_encoder", type=bool, default = False)
-    parser.add_argument("--pretrained_audio_encoder", type = bool, default=False)
-    parser.add_argument("--freeze_audio_encoder", type = bool, default = False)
-    parser.add_argument("--augment_dataset", type = bool, default = True)
+
+    parser.add_argument("--pretrained_image_encoder", action="store_true")
+    parser.add_argument("--freeze_image_encoder", action="store_true")
+    parser.add_argument("--pretrained_audio_encoder", action="store_true")
+    parser.add_argument("--freeze_audio_encoder", action="store_true")
+    parser.add_argument("--augment_dataset", action="store_true")
 
     for key, value in audio_args.items():
         parser.add_argument(f"--{key}", type=type(value), default=value)