forked from wtliao/text2image
-
Notifications
You must be signed in to change notification settings - Fork 0
/
bird.yml
40 lines (31 loc) · 942 Bytes
/
bird.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
CONFIG_NAME: 'bird_sloss01'
DATASET_NAME: 'birds'
DATA_DIR: 'data/birds'
GPU_ID: 0
WORKERS: 1
B_VALIDATION: True # 是否开启验证模式
#B_VALIDATION: False
RESTORE: False
#RESTORE: True # 是否继续训练
loss: 'hinge'
TREE:
BRANCH_NUM: 1
BASE_SIZE: 256
TRAIN:
NF: 64 # default 64
# BATCH_SIZE: 24 #24
BATCH_SIZE: 6
MAX_EPOCH: 600
# when validation, put the path of the trained model here;
NET_G: './saved_model/OneDrive-2022-05-29/fixed/cub/netG_590.pth' #fixed_model
# NET_G: './saved_model/OneDrive-2022-05-29/finetune/cub/netG_550.pth' #pretrain_model
# NET_G: './tmp/bird_sloss01/64/models/netG_595.pth' #my_model
# NET_G: ''
WARMUP_EPOCHS: 100
GSAVE_INTERVAL: 5
DSAVE_INTERVAL: 5
TEXT:
EMBEDDING_DIM: 256
CAPTIONS_PER_IMAGE: 10
# DAMSM_NAME: 'DAMSMencoders/bird/text_encoder200.pth'
DAMSM_NAME: 'saved_model/OneDrive-2022-05-29/finetune/cub/text_encoder_550.pth' #pretrain_model