m

kerem-cekmeceli · Jun 5, 2024 · d582ce0 · d582ce0
1 parent 4ddbc07
commit d582ce0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/MedicalSegmentation/train.py b/MedicalSegmentation/train.py
@@ -39,7 +39,7 @@
 log_the_run = cluster_mode
 
 # Select model type
-model_type = ModelType.UNET  # SEGMENTOR, UNET, SWINUNET, R2ATTNUNET
+model_type = ModelType.SWINUNET  # SEGMENTOR, UNET, SWINUNET, R2ATTNUNET
 
 # Fully Test Time Adaptations (Entropy Minimization)
 ftta = False
@@ -433,8 +433,8 @@
 n_best = 1 if save_checkpoints else 0
 
 if cluster_paths:
-    # ckp_pth = Path('/usr/bmicnas02/data-biwi-01/foundation_models/AllCheckpoints')
-    ckp_pth = main_pth
+    ckp_pth = Path('/usr/bmicnas02/data-biwi-01/foundation_models/AllCheckpoints')
+    # ckp_pth = main_pth
 else:
     ckp_pth = main_pth