cleaning up

2024-08-11 13:33:36 +07:00
parent 62af96e93a
commit a882d53d4e
1 changed files with 2 additions and 1 deletions
@@ -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: