File size: 553 Bytes
c8ddb9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
"""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
|