src/models/modules/__init__.py

"""All the modules used in creation of Generator and Discriminator"""
from .acm import ACM
from .attention import ChannelWiseAttention, SpatialAttention
from .cond_augment import CondAugmentation
from .conv_utils import calc_out_conv, conv1d, conv2d
from .discriminator import Discriminator, WordLevelLogits
from .downsample import down_sample
from .generator import Generator
from .image_encoder import InceptionEncoder, VGGEncoder
from .residual import ResidualBlock
from .text_encoder import TextEncoder
from .upsample import img_up_block, up_sample