File size: 394 Bytes
7e4e601
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

import torch.nn as nn
import torch.nn.functional as F
from torch.nn.utils import spectral_norm
from models.conv_blocks import DownConv
from models.conv_blocks import UpConv
from models.conv_blocks import SeparableConv2D
from models.conv_blocks import InvertedResBlock
from models.conv_blocks import ConvBlock
from utils.common import initialize_weights


class GeneratorV3(nn.Module):
    pass