fix

bluemindai · Mar 4, 2024 · 09f369d · 09f369d
1 parent 53c0baa
commit 09f369d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nnunetv2/experiment_planning/experiment_planners/default_experiment_planner.py b/nnunetv2/experiment_planning/experiment_planners/default_experiment_planner.py
@@ -556,7 +556,7 @@ def plan_experiment(self):
         "spacing": [1.0, 1.0, 1.0],
         "patch_size": [192, 192, 192],
         "UNet_base_num_features": 32,
-        "n_conv_per_stage_encoder": [2, 2, 2, 3, 3, 3],
+        "n_conv_per_stage_encoder": [1, 1, 2, 3, 3, 3],
         "n_conv_per_stage_decoder": [3, 3, 3, 3, 3],
         "num_pool_per_axis": [4, 4, 4],
         "pool_op_kernel_sizes": [
@@ -577,6 +577,7 @@ def plan_experiment(self):
         ],
         "unet_max_num_features": 320,
         "normalization_schemes": ["CTNormalization"],
+        "batch_dice": False,
         "resampling_fn_data": "no_resampling_data_or_seg_to_shape",
         "resampling_fn_seg": "no_resampling_data_or_seg_to_shape",
         "resampling_fn_probabilities": "no_resampling_data_or_seg_to_shape",