kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Import fix for unet

Browse Source
This commit is contained in:
James Betker 2020-11-14 22:09:18 -07:00
parent 5cade6b874
commit 9af049c671
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/archs/stylegan/stylegan2_unet_disc.py
View File

@ -4,6 +4,9 @@ from math import log2
import torch import torch
import torch.nn as nn import torch.nn as nn
from models.archs.stylegan.stylegan2 import attn_and_ff
def leaky_relu(p=0.2): def leaky_relu(p=0.2):
return nn.LeakyReLU(p) return nn.LeakyReLU(p)