File size: 850 Bytes
9622996 |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
DATA:
DATAROOT: /data/CelebA-HQ/
IMG_SIZE: 256
NAME: CelebA-HQ
DATALOADER:
BATCH_SIZE: 12
MICRO_BATCH: 0
NUM_WORKERS: 4
PIN_MEMORY: true
PREFETCH_FACTOR: 2
EVALUATE:
N_EVAL: 0
MASK:
BRUSH_LENGTH_RATIO:
- 0.1
- 0.25
BRUSH_NUM: &id001
- 0
- 0
BRUSH_TURNS:
- 4
- 18
BRUSH_WIDTH_RATIO:
- 0.02
- 0.1
MASK_DIR: /data/NVIDIAIrregularMaskDataset/train/
RECT_LENGTH_RATIO:
- 0.2
- 0.8
RECT_NUM: *id001
MODEL:
N_LAYERS: 8
WEIGHTS: null
SAMPLE:
N_SAMPLES: 100
RANDOM: false
SAVE_DIR: ./samples/
SEED: 1234
TRAIN:
COEF_HOLE: 6.0
COEF_PERCEPTUAL: 0.05
COEF_STYLE: 120
COEF_TV: 0.1
COEF_VALID: 1.0
EVAL_FREQ: 1000
OPTIM:
BETAS:
- 0.5
- 0.999
LR: 0.0002
PRINT_FREQ: 100
RESUME: ./runs/test/ckpt/step039999.pt
SAMPLE_FREQ: 1000
SAVE_FREQ: 5000
TRAIN_STEPS: 200000
|