150 lines
3.5 KiB
JSON
150 lines
3.5 KiB
JSON
{
|
|
"run_name": "p4_3_ddpm_vpred",
|
|
"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_3_ddpm_vpred",
|
|
"noise_schedule": "cosine",
|
|
"pred_type": "v"
|
|
},
|
|
"history": {
|
|
"loss": [
|
|
0.12875029073757494,
|
|
0.08614025376418717,
|
|
0.07806828354572894,
|
|
0.07598056628281234,
|
|
0.07384663673802319,
|
|
0.07230065119030893,
|
|
0.07155742664208524,
|
|
0.07158301252489671,
|
|
0.06912831641319725,
|
|
0.0690550329322871,
|
|
0.06728915909193775,
|
|
0.0680126636965662,
|
|
0.0679081722568625,
|
|
0.06737891206533736,
|
|
0.0675490371341634,
|
|
0.06628662669378468,
|
|
0.06702909948957017,
|
|
0.06577521818889003,
|
|
0.0659722731822831,
|
|
0.06550971675727867,
|
|
0.06532588509572113,
|
|
0.06617079318588616,
|
|
0.06557409760629775,
|
|
0.06460701921779631,
|
|
0.06434725145371552,
|
|
0.06484076265309356,
|
|
0.06488110336005433,
|
|
0.06471904473872776,
|
|
0.06392401659813447,
|
|
0.06467082913423705,
|
|
0.06412526943649237,
|
|
0.06369206279866461,
|
|
0.06399395895532818,
|
|
0.06387244150615655,
|
|
0.06383777140743202,
|
|
0.06308407760742638,
|
|
0.06396170450836165,
|
|
0.06374330889497302,
|
|
0.06341904871420473,
|
|
0.06296812231915119,
|
|
0.06285389444321139,
|
|
0.06360878343256111,
|
|
0.06254984510855542,
|
|
0.06288161397808129,
|
|
0.06314336311899954,
|
|
0.06315476795196788,
|
|
0.06280555519569889,
|
|
0.06323608846013618,
|
|
0.06255563115303078,
|
|
0.06277075296856909,
|
|
0.06311190929104629,
|
|
0.062085086575303323,
|
|
0.061993980668803565,
|
|
0.06232117703104885,
|
|
0.06287038124078868,
|
|
0.06232338945548504,
|
|
0.062193875718447894,
|
|
0.0618046884639905,
|
|
0.06204842089110205,
|
|
0.061992981136800386,
|
|
0.06199613542288032,
|
|
0.06176470703262295,
|
|
0.06138660101235932,
|
|
0.06124315102003578,
|
|
0.06172546414801708,
|
|
0.06145226469056474,
|
|
0.06173897245660042,
|
|
0.06087356748489233,
|
|
0.06146528088670765,
|
|
0.06130675005161355,
|
|
0.06117054966525135,
|
|
0.06046625260168161,
|
|
0.060342767578350685,
|
|
0.06143473189833582,
|
|
0.06048223876163491,
|
|
0.060843704316096425,
|
|
0.0605114245086781,
|
|
0.06107715372410086,
|
|
0.060475895483779095,
|
|
0.06033737308735776,
|
|
0.060130660151505574,
|
|
0.06022480088803503,
|
|
0.06079090322948929,
|
|
0.05941804670370542,
|
|
0.06048406008630991,
|
|
0.0605061265727521,
|
|
0.0595617718819497,
|
|
0.059925066514147654,
|
|
0.05995072215859197,
|
|
0.0600912062307963,
|
|
0.05996395193008531,
|
|
0.05987944811359685,
|
|
0.059719653529489145,
|
|
0.059423190700765856,
|
|
0.05964870073506211,
|
|
0.059674753154763296,
|
|
0.05960713100070373,
|
|
0.05966718835581062,
|
|
0.059533771055822186,
|
|
0.059380597387177825
|
|
],
|
|
"fid": {
|
|
"25": 315.896728515625,
|
|
"50": 160.689453125,
|
|
"75": 81.74005126953125,
|
|
"100": 34.53862762451172
|
|
},
|
|
"train_time_s": 16728.13574409485
|
|
},
|
|
"n_params": 28474755
|
|
} |