From 89f56b20919fa92db0c9e29d67468831f27a0531 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 14 Nov 2020 22:10:45 -0700 Subject: [PATCH] Fix another import --- codes/models/archs/stylegan/stylegan2_unet_disc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/stylegan/stylegan2_unet_disc.py b/codes/models/archs/stylegan/stylegan2_unet_disc.py index 634095b6..2f92750b 100644 --- a/codes/models/archs/stylegan/stylegan2_unet_disc.py +++ b/codes/models/archs/stylegan/stylegan2_unet_disc.py @@ -4,7 +4,7 @@ from math import log2 import torch import torch.nn as nn -from models.archs.stylegan.stylegan2 import attn_and_ff +from models.archs.stylegan.stylegan2 import attn_and_ff, Flatten def leaky_relu(p=0.2):