Fix initialization_kwargs (#147)

creatis-myriad · Nov 16, 2023 · 204d50a · 204d50a
1 parent c5a3baf
commit 204d50a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/ascent/models/components/encoders/convnext.py b/ascent/models/components/encoders/convnext.py
@@ -217,7 +217,11 @@ def __init__(
         # initialize weights
         init_kwargs = {}
         if activation == "leakyrelu":
-            if activation_kwargs is not None and "negative_slope" in activation_kwargs:
+            if (
+                activation_kwargs is not None
+                and "negative_slope" in activation_kwargs
+                and initialization == "kaiming_normal"
+            ):
                 init_kwargs["neg_slope"] = activation_kwargs["negative_slope"]
         self.apply(get_initialization(initialization, **init_kwargs))
 

diff --git a/ascent/models/components/encoders/unet_encoder.py b/ascent/models/components/encoders/unet_encoder.py
@@ -164,7 +164,11 @@ def __init__(
         # initialize weights
         init_kwargs = {}
         if activation == "leakyrelu":
-            if activation_kwargs is not None and "negative_slope" in activation_kwargs:
+            if (
+                activation_kwargs is not None
+                and "negative_slope" in activation_kwargs
+                and initialization == "kaiming_normal"
+            ):
                 init_kwargs["neg_slope"] = activation_kwargs["negative_slope"]
         self.apply(get_initialization(initialization, **init_kwargs))