Updated slice axis and reduced number of epochs.

ivadomed · Dec 22, 2021 · c8359b1 · c8359b1
1 parent 32a1108
commit c8359b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/config/seg_sc.json b/config/seg_sc.json
@@ -30,7 +30,7 @@
             "filter_empty_mask": false,
             "filter_empty_input": false
         },
-        "slice_axis": "axial",
+        "slice_axis": "sagittal",
         "multichannel": false,
         "soft_gt": false
     },
@@ -49,8 +49,8 @@
             "name": "DiceLoss"
         },
         "training_time": {
-            "num_epochs": 500,
-            "early_stopping_patience": 200,
+            "num_epochs": 200,
+            "early_stopping_patience": 50,
             "early_stopping_epsilon": 0.001
         },
         "scheduler": {