Preview of phase 2-5 implementation; needs a full check

This commit is contained in:
Johnny Fernandes
2026-04-30 13:10:33 +01:00
parent 6e32001ebc
commit 7417267117
35 changed files with 3605 additions and 115 deletions
@@ -0,0 +1,19 @@
{
"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
}