compatibility with dynamic_network_architectures

MIC-DKFZ · Mar 1, 2024 · 83dad35 · 83dad35
1 parent 1b5a17d
commit 83dad35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py b/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py
@@ -2,7 +2,7 @@
 from copy import deepcopy
 from typing import Union, List, Tuple
 
-from dynamic_network_architectures.architectures.residual_unet import ResidualEncoderUNet
+from dynamic_network_architectures.architectures.unet import ResidualEncoderUNet
 from dynamic_network_architectures.building_blocks.helper import convert_dim_to_conv_op, get_matching_instancenorm
 from torch import nn
 
@@ -232,4 +232,4 @@ def _keygen(patch_size, strides):
                               n_conv_per_stage_decoder=(1, 1, 1, 1, 1, 1),
                               conv_bias=True, norm_op=nn.InstanceNorm2d, norm_op_kwargs={}, dropout_op=None,
                               nonlin=nn.LeakyReLU, nonlin_kwargs={'inplace': True}, deep_supervision=True)
-    print(net.compute_conv_feature_map_size((512, 512)))  # -> 129793792
+    print(net.compute_conv_feature_map_size((512, 512)))  # -> 129793792