这个是已经训练好的DFGAN,针对CUB的生成器模型,训练轮数601轮
默认bird.yml:
CONFIG_NAME: 'bird'
DATASET_NAME: 'bird'
DATA_DIR: '../data/bird'
GPU_ID: 0
WORKERS: 1
B_VALIDATION: True # True # False
loss: 'hinge'
TREE:
BRANCH_NUM: 1
BASE_SIZE: 256
TRAIN:
NF: 32 # default 64
BATCH_SIZE: 24
MAX_EPOCH: 601
NET_G: '../test'
TEXT:
EMBEDDING_DIM: 256
CAPTIONS_PER_IMAGE: 10
DAMSM_NAME: '../DAMSMencoders/bird/inception/text_encoder200.pth'