CONFIG_NAME: 'attn2'

DATASET_NAME: 'birds'
DATA_DIR: 'data/birds'
GPU_ID: 3
WORKERS: 1

B_VALIDATION: False  # True  # False
TREE:
    BRANCH_NUM: 3


TRAIN:
    FLAG: False
    NET_G: 'models/bird_AttnGAN2.pth'
    B_NET_D: False
    BATCH_SIZE: 100
    NET_E: 'DAMSMencoders/bird/text_encoder200.pth'


GAN:
    DF_DIM: 64
    GF_DIM: 32
    Z_DIM: 100
    R_NUM: 2

TEXT:
    EMBEDDING_DIM: 256
    CAPTIONS_PER_IMAGE: 10
    WORDS_NUM: 25