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

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
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_1_ddpm_linear",
+  "noise_schedule": "linear",
+  "pred_type": "eps"
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_2_ddpm_cosine",
+  "noise_schedule": "cosine",
+  "pred_type": "eps"
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_3_ddpm_vpred",
+  "noise_schedule": "cosine",
+  "pred_type": "v"
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_4_ddpm_wider",
+  "noise_schedule": "cosine",
+  "pred_type": "v",
+  "base_ch": 192,
+  "attn_resolutions": [32, 16, 8]
+}