From 209332292a26649d6a12e2242db2146b5abb951b Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 18 Dec 2020 09:50:41 -0700 Subject: [PATCH] Rosinality stylegan fix --- codes/models/archs/stylegan/stylegan2_rosinality.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/stylegan/stylegan2_rosinality.py b/codes/models/archs/stylegan/stylegan2_rosinality.py index 8a8b8ce6..ba327894 100644 --- 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 )