Context-Encoder-pytorch / celebahq-noadv /config-2023-02-05-02-44-31.yaml
xyfJASON's picture
Upload celebahq-noadv checkpoint and training logs
ccb4de4 verified
raw
history blame contribute delete
555 Bytes
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