Rosinality stylegan fix

2020-12-18 09:50:41 -07:00 · 2020-12-18 09:50:41 -07:00 · 209332292a
commit 209332292a
parent d875ca8342
1 changed files with 1 additions and 1 deletions
--- a/codes/models/archs/stylegan/stylegan2_rosinality.py
+++ b/codes/models/archs/stylegan/stylegan2_rosinality.py
@ -33,7 +33,7 @@ def fused_leaky_relu(input, bias=None, negative_slope=0.2, scale=2 ** 0.5):
        rest_dim = [1] * (input.ndim - bias.ndim - 1)
        return (
            F.leaky_relu(
-                input + bias.view(1, bias.shape[0], *rest_dim), negative_slope=0.2
+                input + bias.view(1, bias.shape[0], *rest_dim), negative_slope=negative_slope
            )
            * scale
        )