File size: 555 Bytes
ccb4de4 |
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 |
DATA:
DATAROOT: /data/CelebA-HQ/
IMG_SIZE: 128
NAME: CelebA-HQ
DATALOADER:
BATCH_SIZE: 256
MICRO_BATCH: 0
NUM_WORKERS: 8
PIN_MEMORY: true
PREFETCH_FACTOR: 4
EVALUATE:
N_EVAL: 0
MODEL:
WEIGHTS: null
SAMPLE:
N_SAMPLES: 100
RANDOM: false
SAVE_DIR: ./samples/
SEED: 1234
TRAIN:
EVAL_FREQ: 400
LAMBDA_ADV: 0.0
LAMBDA_REC: 0.999
OPTIM_D:
BETAS: &id001
- 0.5
- 0.999
LR: 0.0002
OPTIM_G:
BETAS: *id001
LR: 0.002
PRINT_FREQ: 100
RESUME: null
SAMPLE_FREQ: 400
SAVE_FREQ: 5000
TRAIN_STEPS: 25000
|