fix #2425

MIC-DKFZ · Aug 30, 2024 · 80cd16c · 80cd16c
1 parent cbeaa03
commit 80cd16c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnunetv2/utilities/get_network_from_plans.py b/nnunetv2/utilities/get_network_from_plans.py
@@ -28,8 +28,8 @@ def get_network_from_plans(arch_class_name, arch_kwargs, arch_kwargs_req_import,
         else:
             raise ImportError('Network class could not be found, please check/correct your plans file')
 
-    if deep_supervision is not None and 'deep_supervision' not in arch_kwargs.keys():
-        arch_kwargs['deep_supervision'] = deep_supervision
+    if deep_supervision is not None:
+        architecture_kwargs['deep_supervision'] = deep_supervision
 
     network = nw_class(
         input_channels=input_channels,