Results from phase 2 to 4

2026-05-01 21:08:42 +00:00
parent 14aed699ed
commit 1a7f67ab9c
203 changed files with 4524 additions and 0 deletions
@@ -0,0 +1,150 @@
+{
+  "run_name": "p4_2_ddpm_cosine",
+  "config": {
+    "batch_size": 64,
+    "ema_decay": 0.9999,
+    "data_dir": "cropped/generator",
+    "sources": [
+      "wiki"
+    ],
+    "subsample": 1.0,
+    "sample_interval": 10,
+    "fid_interval": 25,
+    "fid_n_real": 5000,
+    "epochs": 100,
+    "augment": "hflip",
+    "image_size": 64,
+    "model": "ddpm",
+    "T": 1000,
+    "ddim_steps": 100,
+    "lr": 0.0002,
+    "base_ch": 128,
+    "ch_mult": [
+      1,
+      2,
+      2,
+      2
+    ],
+    "attn_resolutions": [
+      16,
+      8
+    ],
+    "num_res_blocks": 2,
+    "dropout": 0.1,
+    "run_name": "p4_2_ddpm_cosine",
+    "noise_schedule": "cosine",
+    "pred_type": "eps"
+  },
+  "history": {
+    "loss": [
+      0.0745811819847132,
+      0.042058256996047295,
+      0.038731603282225184,
+      0.0365341946355298,
+      0.03596278707870943,
+      0.03499822471784348,
+      0.034039977941916794,
+      0.03339788123853823,
+      0.033659401990505114,
+      0.0328750506353875,
+      0.03281905785656701,
+      0.032175978936223135,
+      0.03264562071213483,
+      0.03167968711011812,
+      0.03252785405717217,
+      0.032265412987361096,
+      0.032163158218320616,
+      0.03181939549409808,
+      0.03145227379476031,
+      0.031572662918772705,
+      0.0314074662896112,
+      0.03058532617874762,
+      0.031417921515834384,
+      0.030867751819105484,
+      0.031200444417344965,
+      0.030372575657744694,
+      0.031070002880042944,
+      0.0306555292624821,
+      0.031143788686101764,
+      0.030448843175784137,
+      0.03087240754841612,
+      0.03053835191580857,
+      0.030471389926174004,
+      0.0302293243435904,
+      0.030659112543997984,
+      0.030551793045786202,
+      0.030635309206624318,
+      0.030368149071597517,
+      0.030187387226356402,
+      0.03037746619974446,
+      0.030544078081018396,
+      0.0304682625648685,
+      0.030033117888543088,
+      0.030058689524905167,
+      0.030037028862274866,
+      0.02957952331799345,
+      0.03012158986356141,
+      0.029973779056762528,
+      0.030247858279726952,
+      0.02959363942599704,
+      0.029731284608093336,
+      0.03002101383928178,
+      0.029881187225094974,
+      0.029322936573726498,
+      0.029073657284880806,
+      0.02956077633385793,
+      0.029618583959885515,
+      0.029847813658734672,
+      0.029749778817750067,
+      0.02967120576092703,
+      0.02939097446381536,
+      0.029562414946177833,
+      0.02957131928947364,
+      0.029576030472277574,
+      0.029486760393612914,
+      0.028896253977894273,
+      0.029079640140900247,
+      0.02946651985661851,
+      0.02956312265382427,
+      0.028787078188620824,
+      0.029077194975370653,
+      0.02904141321579297,
+      0.02869541201995224,
+      0.029384889960304923,
+      0.029296722736719072,
+      0.028805798423102397,
+      0.029137278389599588,
+      0.02882065039733027,
+      0.028110843072207566,
+      0.029165515112969227,
+      0.02838028352269823,
+      0.028928894915769242,
+      0.02887988943232494,
+      0.02871215812718639,
+      0.02849309494257228,
+      0.02836313749400851,
+      0.02900838313433222,
+      0.028941123808025677,
+      0.028982899480093367,
+      0.028617646975012925,
+      0.028945480441499468,
+      0.028021262546714682,
+      0.028656914595387176,
+      0.028812512554403428,
+      0.02845827827994258,
+      0.02854005663265657,
+      0.028420659669666972,
+      0.028380179300140113,
+      0.0278120991280382,
+      0.0285125195613911
+    ],
+    "fid": {
+      "25": 249.73973083496094,
+      "50": 282.24053955078125,
+      "75": 239.8434295654297,
+      "100": 132.29640197753906
+    },
+    "train_time_s": 15527.574454307556
+  },
+  "n_params": 28474755
+}