From 656f5d169d21b0a7b099a485090263d463f03bd4 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Wed, 24 Aug 2022 10:20:23 +0200
Subject: [PATCH] again update for issue with act name
---
mlair/model_modules/residual_networks.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/mlair/model_modules/residual_networks.py b/mlair/model_modules/residual_networks.py
index 8d39d776..a9b502c4 100644
--- a/mlair/model_modules/residual_networks.py
+++ b/mlair/model_modules/residual_networks.py
@@ -47,7 +47,10 @@ 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.args[0] if isinstance(act, partial) else act.__name__
+ if isinstance(act, partial):
+ act_name = act.args[0] if act.func.__name__ == "Activation" else act.func.__name__
+ else:
+ act_name = act.__name__
use_1x1conv = layer_kwargs.pop("use_1x1conv", False)
def block(x):
--
GitLab