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_1_ddpm_linear",
+  "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_1_ddpm_linear",
+    "noise_schedule": "linear",
+    "pred_type": "eps"
+  },
+  "history": {
+    "loss": [
+      0.050263571692232646,
+      0.02514873994474546,
+      0.02225509669400879,
+      0.021068911675093137,
+      0.02032521784775214,
+      0.019192529312916037,
+      0.018796856887240566,
+      0.019092055869439982,
+      0.018012497312405236,
+      0.018579239720223933,
+      0.018120780589186355,
+      0.01804902349025584,
+      0.017445260646920174,
+      0.0174015331078066,
+      0.01720998039803444,
+      0.017182101499543995,
+      0.017374234542680476,
+      0.01719019105896736,
+      0.017258047804825455,
+      0.017472001728920154,
+      0.01717184852155992,
+      0.017268840462351456,
+      0.016940360013236348,
+      0.016780947783810653,
+      0.016770626064071543,
+      0.01686398620601807,
+      0.0171233544496294,
+      0.016849436540490925,
+      0.01643650009173454,
+      0.016715853648164716,
+      0.01591222248792362,
+      0.01663316461122316,
+      0.017018022838558078,
+      0.016728110333832983,
+      0.01637174110983013,
+      0.01640069789977537,
+      0.016222019855959866,
+      0.01619866752348299,
+      0.016273524062946822,
+      0.016225511587272663,
+      0.016525169477288604,
+      0.016554634154447887,
+      0.01636095449495583,
+      0.016358451291910794,
+      0.01612802425849164,
+      0.016018457704772934,
+      0.015985010074387886,
+      0.015972342274478104,
+      0.01617517040915087,
+      0.016143278803867407,
+      0.016039738931064296,
+      0.016184791375317763,
+      0.01608435499171416,
+      0.016031058468362395,
+      0.016189519307003,
+      0.016219924015000973,
+      0.016112841876685366,
+      0.015630949461737122,
+      0.015760562722929396,
+      0.015901298478094496,
+      0.01586071201440934,
+      0.016167635994597186,
+      0.015887681410735488,
+      0.01600054983431712,
+      0.015733229437182285,
+      0.015523023833719712,
+      0.01581668923011957,
+      0.015652269207569968,
+      0.015604886598487059,
+      0.015797711234205425,
+      0.015701896441368084,
+      0.015930752163458392,
+      0.015848295793980043,
+      0.01567347500568781,
+      0.015367114703513237,
+      0.015625674345204208,
+      0.01559769835534036,
+      0.015479316330976529,
+      0.015648531339441735,
+      0.015174523765085917,
+      0.015752060055279005,
+      0.015384152635104127,
+      0.015699152740188196,
+      0.015053038565943448,
+      0.015501487074801937,
+      0.015345309023212036,
+      0.015243542788582098,
+      0.01545036704534203,
+      0.01533815077541826,
+      0.015173954715052795,
+      0.015228697487837674,
+      0.01517730934294657,
+      0.015363585634722223,
+      0.015174964277678702,
+      0.01534465596310667,
+      0.015116462827278063,
+      0.015595124076263836,
+      0.015289406492136037,
+      0.015195867164124154,
+      0.015012384551521549
+    ],
+    "fid": {
+      "25": 333.25946044921875,
+      "50": 311.42364501953125,
+      "75": 255.9425506591797,
+      "100": 134.49986267089844
+    },
+    "train_time_s": 15595.587987422943
+  },
+  "n_params": 28474755
+}