Context-Encoder-pytorch / celebahq /config-2023-02-04-11-41-44.yaml
xyfJASON's picture
Upload celebahq checkpoint and training logs
fd0ccfc verified
raw
history blame contribute delete
557 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: 2
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.001
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: 40000