DRL_PROJ/generator/outputs/logs/p5_ddpm.json

{
  "run_name": "p5_ddpm",
  "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,
    "num_workers": 2,
    "run_name": "p5_ddpm",
    "model": "ddpm",
    "epochs": 1000,
    "augment": "hflip",
    "image_size": 64,
    "T": 1000,
    "noise_schedule": "cosine",
    "pred_type": "v",
    "base_ch": 192,
    "ch_mult": [
      1,
      2,
      2,
      2
    ],
    "attn_resolutions": [
      32,
      16,
      8
    ],
    "num_res_blocks": 2,
    "dropout": 0.1,
    "lr": 0.0002,
    "ddim_steps": 100,
    "max_train_hours": 24,
    "fid_patience": 2
  },
  "history": {
    "loss": [
      0.1361132180557037,
      0.08347086386325267,
      0.07713027890676107,
      0.0735087690827174,
      0.07138374625331062,
      0.07048111248952456,
      0.0699430058001835,
      0.06959152534508552,
      0.0687942563309374,
      0.06749710015570506,
      0.06797479636744301,
      0.06707314404253012,
      0.06645634254583946,
      0.06616808988281295,
      0.06632597825657098,
      0.06543161581533077,
      0.06567037307744862,
      0.06560073971238911,
      0.06487136873870324,
      0.06486488640722302,
      0.06474818710365102,
      0.0648785232542417,
      0.06409876514226198,
      0.06419924900540684,
      0.06415374198148394,
      0.06415291013530432,
      0.06390660826880963,
      0.06351337538888821,
      0.0637402735157018,
      0.06242381525026937,
      0.0636787190157761,
      0.06266482662345864,
      0.0638032130514964,
      0.06235016199449698,
      0.06285461769081079,
      0.06259892232970804,
      0.06228241114280163,
      0.06305973536661293,
      0.06274244885764316,
      0.062076514268405415,
      0.06195374166704396,
      0.06290199403834139,
      0.06198460275195857,
      0.0624536736908122,
      0.06327703114376108,
      0.061569202508236096,
      0.0628280364550077,
      0.06169611253003534,
      0.06185008156407847,
      0.06172168297040412,
      0.061782848997375905,
      0.06151117438561896,
      0.061827609793115884,
      0.061956947136065386,
      0.06214875884704355,
      0.062044317284837745,
      0.061401939210601345,
      0.06086002392933155,
      0.061775257596029684,
      0.061602149421396926,
      0.06117732557030315,
      0.061202412471175194,
      0.061622718205818765,
      0.06128816264402917,
      0.0614978774315399,
      0.061346558766423635,
      0.06128153991368082,
      0.06116244996077994,
      0.06111229184823923,
      0.06107063669488471,
      0.06084184085106493,
      0.060991186720247455,
      0.06110310762260969,
      0.060644112224889614,
      0.06105265081024323,
      0.060728371310501524,
      0.0612431171660622,
      0.06059174603408473,
      0.06193933507793734,
      0.06114670591285595,
      0.06008413203187987,
      0.060815918625483655,
      0.06038160178746678,
      0.060638460115744516,
      0.06106071204392829,
      0.060512709447270274,
      0.05989936954126908,
      0.06053547763353229,
      0.060993485152721405,
      0.06054355477134132,
      0.060113425056139626,
      0.060656892772541084,
      0.06028386112302542,
      0.06064964496554472,
      0.060564143551338434,
      0.06023223410782396,
      0.0606433203141404,
      0.06019309561094667,
      0.060663971755430736,
      0.06061358017544461,
      0.06051926234434558,
      0.060519506510060564,
      0.05998779779188653,
      0.060273363700725585,
      0.060976108679404624,
      0.05960293089509265,
      0.06006794761165849,
      0.06001491510333159,
      0.059924575054429025,
      0.060125686371555694,
      0.06028291987430336,
      0.06011252240556428,
      0.0603115351504495,
      0.059780152243936166,
      0.05937349914899494,
      0.060589286212164625,
      0.05928194311917083,
      0.0603618670700707,
      0.05988672448115216,
      0.06036584552886904,
      0.05977793929413853,
      0.0594497075320309,
      0.059364209859034955,
      0.059953983116926804,
      0.059721899139257066,
      0.06011831242209061,
      0.059677510808866754,
      0.059987764399594225,
      0.060053561774329245,
      0.059788390898551695,
      0.05936959649348616,
      0.059900946533068634,
      0.05933099720053948,
      0.05939908983169967,
      0.0593241080084545,
      0.05961700674528495,
      0.05928131563859618,
      0.05942184386504257,
      0.060225125601212695,
      0.0589471840673787,
      0.06034273053240827,
      0.05934861272923712,
      0.05955804113107614,
      0.05946602335629555,
      0.05946855390301117,
      0.05917746134293385,
      0.05913895009579057,
      0.060033635260202944,
      0.05908472824873578,
      0.060041244786519274,
      0.05998928384043467,
      0.059578305993897795,
      0.05958271934849839,
      0.05902576439369183,
      0.06004863530875017,
      0.058922704555985764,
      0.059662471589051254,
      0.05893384579282541,
      0.05959883256632294,
      0.05884528258003485,
      0.05999988574598335,
      0.059614969194572195,
      0.05979255826459227,
      0.05972598887916304,
      0.05883011880179501,
      0.05925349895165771,
      0.059959421220880285,
      0.05954511561550391,
      0.05976989628890386,
      0.059382665185974196,
      0.059384163946677476,
      0.059122750774408,
      0.05936474042634169,
      0.059725822403262824,
      0.05970647112012674
    ],
    "fid": {
      "25": 312.4023742675781,
      "50": 174.07015991210938,
      "75": 81.27722930908203,
      "100": 34.90231704711914,
      "125": 19.03875160217285,
      "150": 19.847490310668945,
      "175": 20.106101989746094
    },
    "train_time_s": 40025.90805530548
  },
  "n_params": 64770819
}