Johnny Fernandes
548 lines
11 KiB
JSON
548 lines
11 KiB
JSON
{
|
|
"run_name": "p3_2_vae_perceptual",
|
|
"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,
|
|
"epochs": 100,
|
|
"augment": "hflip",
|
|
"image_size": 64,
|
|
"model": "vae",
|
|
"latent_dim": 256,
|
|
"ngf": 64,
|
|
"grad_clip": 1.0,
|
|
"run_name": "p3_2_vae_perceptual",
|
|
"lr": 0.0005,
|
|
"beta_kl": 0.25,
|
|
"lambda_perceptual": 0.1,
|
|
"lambda_adversarial": 0.0
|
|
},
|
|
"history": {
|
|
"recon_loss": [
|
|
0.07014625494042014,
|
|
0.04432385861396025,
|
|
0.03902342810462683,
|
|
0.036902043435117625,
|
|
0.035906293887135565,
|
|
0.035914984559560686,
|
|
0.0360899488401846,
|
|
0.036847473583860785,
|
|
0.037394630213260144,
|
|
0.03836990671200503,
|
|
0.039295883698022775,
|
|
0.040212508386526354,
|
|
0.04124497003757801,
|
|
0.04212733724305772,
|
|
0.04325710433638758,
|
|
0.04402907300963361,
|
|
0.045103192679647706,
|
|
0.04595246975525068,
|
|
0.04698181097419598,
|
|
0.047825980820080154,
|
|
0.04772546164627768,
|
|
0.04763130701950982,
|
|
0.047709369705591954,
|
|
0.04753121634961194,
|
|
0.04741841174948674,
|
|
0.04729832872016053,
|
|
0.04722155279551561,
|
|
0.0469712430340612,
|
|
0.04693082829093576,
|
|
0.0468663705401441,
|
|
0.04660685717040657,
|
|
0.04661381538384236,
|
|
0.046462044382515624,
|
|
0.0464134263431924,
|
|
0.046272445438254595,
|
|
0.04624118105882508,
|
|
0.046066727958874315,
|
|
0.04603223238363225,
|
|
0.04589838094404365,
|
|
0.0458371472489248,
|
|
0.04577782691225537,
|
|
0.04567820464985238,
|
|
0.04560983914913785,
|
|
0.045465615761076286,
|
|
0.04543259674603613,
|
|
0.04530587481159685,
|
|
0.04528731873465909,
|
|
0.04525673297098559,
|
|
0.04507890154217553,
|
|
0.0450239604514124,
|
|
0.04497251241730574,
|
|
0.044865927300774135,
|
|
0.04475303162207715,
|
|
0.044664276763796806,
|
|
0.044507257990602754,
|
|
0.0444452501515038,
|
|
0.044273821509674065,
|
|
0.044196275588220514,
|
|
0.044097925846775375,
|
|
0.04397543971864586,
|
|
0.04393626836279773,
|
|
0.043823353540247835,
|
|
0.04371356347209623,
|
|
0.043580674797169164,
|
|
0.04348461520181507,
|
|
0.04339473670682846,
|
|
0.043282039009798795,
|
|
0.04320543994888281,
|
|
0.04314021340324583,
|
|
0.04300904847904403,
|
|
0.04293948887950844,
|
|
0.04286929544730064,
|
|
0.04278319672896312,
|
|
0.04270272867547141,
|
|
0.0425933551393513,
|
|
0.04250210111276207,
|
|
0.04243518800562263,
|
|
0.04233357961424905,
|
|
0.04227217231105026,
|
|
0.042201977119677596,
|
|
0.04207655237430436,
|
|
0.04206944097024508,
|
|
0.0420022471768097,
|
|
0.041929350385808535,
|
|
0.04179893332159417,
|
|
0.041740718671781384,
|
|
0.04166606991177695,
|
|
0.04158131634959808,
|
|
0.04155632069445828,
|
|
0.04140628448440733,
|
|
0.041379960253834724,
|
|
0.041310225080093764,
|
|
0.04123378162168794,
|
|
0.04118641266105776,
|
|
0.04113346862837545,
|
|
0.041023712032116376,
|
|
0.04102735277902112,
|
|
0.04096395063062764,
|
|
0.04094440599059702,
|
|
0.040864359348630294
|
|
],
|
|
"kl_loss": [
|
|
0.9578518337673612,
|
|
0.8413855069213443,
|
|
0.7201743645545764,
|
|
0.6246620782165446,
|
|
0.5550231880102402,
|
|
0.49975109450582766,
|
|
0.454662803783376,
|
|
0.4155570190941167,
|
|
0.3821740793621438,
|
|
0.352426735827556,
|
|
0.32639839258204156,
|
|
0.3032199253893306,
|
|
0.2829984358360625,
|
|
0.26505850424241817,
|
|
0.24903864537676176,
|
|
0.23453107457130384,
|
|
0.22166774176761636,
|
|
0.21033292894180006,
|
|
0.19950743799662998,
|
|
0.18994869887192026,
|
|
0.18805619294189999,
|
|
0.18697467955768618,
|
|
0.1855154659312505,
|
|
0.1852042447680082,
|
|
0.18474761941112006,
|
|
0.1843619988514827,
|
|
0.18399313028551575,
|
|
0.18352704119478536,
|
|
0.18345636556036451,
|
|
0.18366932483692455,
|
|
0.18345450644946507,
|
|
0.18317942117523944,
|
|
0.1830864556006387,
|
|
0.1831519056079734,
|
|
0.18329900547734693,
|
|
0.1829878402252992,
|
|
0.18321112291807803,
|
|
0.18296580338197896,
|
|
0.18312124166096377,
|
|
0.18301436457878503,
|
|
0.1831260528574642,
|
|
0.18320244878657863,
|
|
0.18332607687538505,
|
|
0.18335427452101666,
|
|
0.18322811696009758,
|
|
0.18336237903334138,
|
|
0.183573446308191,
|
|
0.18356411601615769,
|
|
0.18363770969912538,
|
|
0.18377095862076834,
|
|
0.18369814056234482,
|
|
0.18386876185098264,
|
|
0.18424720492245805,
|
|
0.18420857423518458,
|
|
0.18395239082921264,
|
|
0.18442093539569113,
|
|
0.1842692175036312,
|
|
0.18475162495787328,
|
|
0.18502490509014863,
|
|
0.18480629139603713,
|
|
0.1851997250993537,
|
|
0.1850732435973791,
|
|
0.1854018302809479,
|
|
0.18547267151566652,
|
|
0.18569090538936803,
|
|
0.1859345402664099,
|
|
0.18615943276219898,
|
|
0.18631401260057065,
|
|
0.1865774132948146,
|
|
0.1867309583303256,
|
|
0.18672093723574254,
|
|
0.18682665680336136,
|
|
0.18679539349853483,
|
|
0.18683723016427115,
|
|
0.1874776411578696,
|
|
0.18746165714712223,
|
|
0.1873003001141752,
|
|
0.18763184111215112,
|
|
0.18799093031348327,
|
|
0.18802655252635989,
|
|
0.18824852079662502,
|
|
0.1882636730169129,
|
|
0.18849152215143555,
|
|
0.1886035389561429,
|
|
0.188697873813729,
|
|
0.18883028218888828,
|
|
0.1889004738858113,
|
|
0.18922981174073666,
|
|
0.18946354345888153,
|
|
0.18955816268029377,
|
|
0.18944057507010606,
|
|
0.18984314531852037,
|
|
0.19001257871715432,
|
|
0.19001863161340737,
|
|
0.18997256346365327,
|
|
0.19019658920856622,
|
|
0.19018171425176483,
|
|
0.19038618680758354,
|
|
0.19034281325263855,
|
|
0.19046620505615178
|
|
],
|
|
"perc_loss": [
|
|
3.257610213552785,
|
|
2.987424520855276,
|
|
2.8892488026211405,
|
|
2.8424616914529066,
|
|
2.817538415774321,
|
|
2.807587738220508,
|
|
2.8047564518757357,
|
|
2.8080830421203222,
|
|
2.8133585376617236,
|
|
2.821376688969441,
|
|
2.830571848612565,
|
|
2.840076332927769,
|
|
2.8507282504668603,
|
|
2.8593530094521675,
|
|
2.8690790253826695,
|
|
2.877339167982085,
|
|
2.8865632745954724,
|
|
2.8936263704911256,
|
|
2.9023511684857883,
|
|
2.9098138376178904,
|
|
2.906375333794162,
|
|
2.9055831661591163,
|
|
2.9031203985214233,
|
|
2.9002673641229286,
|
|
2.897312533651662,
|
|
2.8946053095352955,
|
|
2.8919762974111443,
|
|
2.8902457397208257,
|
|
2.8877664065768576,
|
|
2.884983114197723,
|
|
2.882109224286854,
|
|
2.8805122767758164,
|
|
2.8777520192994013,
|
|
2.876453428186922,
|
|
2.874014714334765,
|
|
2.8723304679251123,
|
|
2.870547831058502,
|
|
2.86905308564504,
|
|
2.8669193524580736,
|
|
2.865350999383845,
|
|
2.8633312608441734,
|
|
2.861651761409564,
|
|
2.860612149421985,
|
|
2.858717398256318,
|
|
2.857239680412488,
|
|
2.8558127604998074,
|
|
2.853911985189487,
|
|
2.852791876364977,
|
|
2.850916815109742,
|
|
2.8496591963319697,
|
|
2.8490061500133614,
|
|
2.8465628389619355,
|
|
2.844754463587052,
|
|
2.842957689211919,
|
|
2.840948245464227,
|
|
2.8397679522506194,
|
|
2.837174654006958,
|
|
2.835211678957328,
|
|
2.833740895629948,
|
|
2.8325765499701867,
|
|
2.830295750218579,
|
|
2.8292511431579914,
|
|
2.8273453595291853,
|
|
2.8257394835480256,
|
|
2.823762384744791,
|
|
2.822507558215378,
|
|
2.8205288745399213,
|
|
2.819065808230995,
|
|
2.8176180297492914,
|
|
2.8165967805772767,
|
|
2.8143360721759305,
|
|
2.8135039892971005,
|
|
2.8120578970664587,
|
|
2.8098325041624217,
|
|
2.8090034366672874,
|
|
2.8076502920215964,
|
|
2.80601545684358,
|
|
2.805327193859296,
|
|
2.803745285058633,
|
|
2.802450499473474,
|
|
2.800580952412043,
|
|
2.799968459157862,
|
|
2.798532901156662,
|
|
2.797372330967178,
|
|
2.795650184663952,
|
|
2.79500452371744,
|
|
2.79342931152409,
|
|
2.792220654650631,
|
|
2.7911063600809145,
|
|
2.7897617689564695,
|
|
2.7887179505111823,
|
|
2.7880892407180915,
|
|
2.7868736916118197,
|
|
2.785768971993373,
|
|
2.784949649602939,
|
|
2.784032260760283,
|
|
2.783316181765662,
|
|
2.7821382457374506,
|
|
2.7815805586994204,
|
|
2.780150015639444
|
|
],
|
|
"adv_g_loss": [
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0
|
|
],
|
|
"adv_d_loss": [
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0
|
|
],
|
|
"fid": {
|
|
"25": 85.4538345336914,
|
|
"50": 70.30448150634766,
|
|
"75": 68.88232421875,
|
|
"100": 68.23878479003906
|
|
},
|
|
"train_time_s": 1526.8077104091644
|
|
},
|
|
"n_params": 10608451
|
|
} |