Johnny Fernandes
20 lines
373 B
JSON
20 lines
373 B
JSON
{
|
|
"epochs": 100,
|
|
"data_dir": "cropped/generator",
|
|
"sources": ["wiki"],
|
|
"augment": "hflip",
|
|
"image_size": 64,
|
|
"model": "ddpm",
|
|
"T": 1000,
|
|
"ddim_steps": 100,
|
|
"lr": 2e-4,
|
|
"base_ch": 128,
|
|
"ch_mult": [1, 2, 2, 2],
|
|
"attn_resolutions": [16, 8],
|
|
"num_res_blocks": 2,
|
|
"dropout": 0.1,
|
|
"sample_interval": 10,
|
|
"fid_interval": 25,
|
|
"fid_n_real": 5000
|
|
}
|