GAN_AI_hw5 / training_config.json
yuting89830's picture
Upload 6 files
e64707c verified
Raw
History Blame Contribute Delete
456 Bytes
{
"image_dir": "public_data/images",
"output_dir": "steps50000_bs16_lr1e-4_seed42_no_inf100_inferbs32_ema",
"image_size": 256,
"batch_size": 16,
"num_epochs": 100,
"max_steps": 50000,
"lr": 0.0001,
"save_every": 5000,
"sample_every": 1000,
"num_preview_samples": 4,
"preview_inference_steps": 30,
"resume_from_checkpoint": null,
"seed": 42,
"num_workers": 4,
"mixed_precision": "no",
"ema_decay": 0.9999,
"no_ema": false
}