150 lines
3.6 KiB
JSON
150 lines
3.6 KiB
JSON
{
|
|
"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
|
|
} |