diff --git a/mlair/model_modules/residual_networks.py b/mlair/model_modules/residual_networks.py index 2c8f3cfbe9c1f413ef8c7af3e1dc04f164092fed..8d39d776238341fbf2da29673910a2127d15528b 100644 --- a/mlair/model_modules/residual_networks.py +++ b/mlair/model_modules/residual_networks.py @@ -1,6 +1,7 @@ __author__ = "Lukas Leufen" __date__ = "2021-08-23" +from functools import partial from mlair.model_modules.branched_input_networks import BranchedInputCNN @@ -46,7 +47,7 @@ class BranchedInputResNet(BranchedInputCNN): layer_name = layer_kwargs.pop("name").split("_") layer_name = "_".join([*layer_name[0:2], "%s", *layer_name[2:]]) act = layer_kwargs.pop("activation") - act_name = act.__name__ + act_name = act.args[0] if isinstance(act, partial) else act.__name__ use_1x1conv = layer_kwargs.pop("use_1x1conv", False) def block(x):