google-research
11 строк · 590.0 Байт
1--- external/eg3d/training/dual_discriminator.py 2023-04-06 03:56:53.605414776 +0000
2+++ external_reference/eg3d/training/dual_discriminator.py 2023-04-06 03:41:04.690731872 +0000
3@@ -15,7 +15,7 @@
4import torch
5from torch_utils import persistence
6from torch_utils.ops import upfirdn2d
7-from training.networks_stylegan2 import DiscriminatorBlock, MappingNetwork, DiscriminatorEpilogue
8+from external.stylegan.training.networks_stylegan2_terrain import DiscriminatorBlock, MappingNetwork, DiscriminatorEpilogue
9
10@persistence.persistent_class
11class SingleDiscriminator(torch.nn.Module):
12