From a882d53d4e5a15425235fcda2c5ed4fdc550f3fe Mon Sep 17 00:00:00 2001 From: FuouM Date: Sun, 11 Aug 2024 13:33:36 +0700 Subject: [PATCH] cleaning up --- module_propih/networks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/module_propih/networks.py b/module_propih/networks.py index bc3be00..ffd5e82 100644 --- a/module_propih/networks.py +++ b/module_propih/networks.py @@ -215,7 +215,8 @@ def get_norm_layer(norm_type="instance"): nn.InstanceNorm2d, affine=False, track_running_stats=False ) elif norm_type == "none": - norm_layer = lambda x: Identity() + def norm_layer(x): + return Identity() # elif norm_type.startswith('rain'): # norm_layer = RAIN else: