mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rosinality stylegan fix

Browse Source
This commit is contained in:
James Betker 2020-12-18 09:50:41 -07:00
parent d875ca8342
commit 209332292a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/archs/stylegan/stylegan2_rosinality.py
View File

@ -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) rest_dim = [1] * (input.ndim - bias.ndim - 1)
return ( return (
F.leaky_relu( 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 * scale
) )