celebahq-256/config-2023-02-07-09-27-42.yaml

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