FilesExpand file tree

 master

/

coco_s2.yml

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

30 lines (27 loc) · 499 Bytes

 master

/

coco_s2.yml

Copy path

Top

File metadata and controls

• Code
• Blame

30 lines (27 loc) · 499 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

CONFIG_NAME: 'stageII'

DATASET_NAME: 'coco'

EMBEDDING_TYPE: 'cnn-rnn'

GPU_ID: '0,1'

Z_DIM: 100

STAGE1_G: '../output/coco_stageI/Model/netG_epoch_120.pth'

DATA_DIR: '../data/coco'

WORKERS: 4

IMSIZE: 256

STAGE: 2

TRAIN:

FLAG: True

BATCH_SIZE: 40

MAX_EPOCH: 100

LR_DECAY_EPOCH: 20

SNAPSHOT_INTERVAL: 5

DISCRIMINATOR_LR: 0.0002

GENERATOR_LR: 0.0002

COEFF:

KL: 2.0

GAN:

CONDITION_DIM: 128

DF_DIM: 96

GF_DIM: 192

R_NUM: 2

TEXT:

DIMENSION: 1024