Johnny Fernandes
25 lines
498 B
JSON
25 lines
498 B
JSON
{
|
|
"run_name": "p5_ddpm",
|
|
"model": "ddpm",
|
|
"epochs": 400,
|
|
"data_dir": "cropped/generator",
|
|
"sources": ["wiki"],
|
|
"augment": "hflip",
|
|
"image_size": 64,
|
|
"T": 1000,
|
|
"noise_schedule": "cosine",
|
|
"pred_type": "v",
|
|
"base_ch": 192,
|
|
"ch_mult": [1, 2, 2, 2],
|
|
"attn_resolutions": [32, 16, 8],
|
|
"num_res_blocks": 2,
|
|
"dropout": 0.1,
|
|
"lr": 2e-4,
|
|
"ddim_steps": 100,
|
|
"sample_interval": 10,
|
|
"fid_interval": 25,
|
|
"fid_n_real": 5000,
|
|
"max_train_hours": 24,
|
|
"fid_patience": 2
|
|
}
|