From 09f369dca0b3d73bb2ffa0474c39895d12cff1c6 Mon Sep 17 00:00:00 2001 From: eolika Date: Mon, 4 Mar 2024 23:05:53 +0300 Subject: [PATCH] fix --- .../experiment_planners/default_experiment_planner.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nnunetv2/experiment_planning/experiment_planners/default_experiment_planner.py b/nnunetv2/experiment_planning/experiment_planners/default_experiment_planner.py index fb1df8274..d930b0578 100644 --- 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",