Spaces:
Build error
Build error
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 |