Import fix for unet

2020-11-14 22:09:18 -07:00 · 2020-11-14 22:09:18 -07:00 · 9af049c671
commit 9af049c671
parent 5cade6b874
1 changed files with 3 additions and 0 deletions
--- a/codes/models/archs/stylegan/stylegan2_unet_disc.py
+++ b/codes/models/archs/stylegan/stylegan2_unet_disc.py
@ -4,6 +4,9 @@ from math import log2
 import torch
 import torch.nn as nn

+from models.archs.stylegan.stylegan2 import attn_and_ff
+
+
 def leaky_relu(p=0.2):
    return nn.LeakyReLU(p)