Johnny Fernandes
550 lines
13 KiB
JSON
550 lines
13 KiB
JSON
{
|
|
"run_name": "p3_3_vae_patchgan",
|
|
"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_3_vae_patchgan",
|
|
"lr": 0.0005,
|
|
"lr_d": 0.0001,
|
|
"beta_kl": 0.25,
|
|
"lambda_perceptual": 0.1,
|
|
"lambda_adversarial": 0.01,
|
|
"ndf_patch": 64
|
|
},
|
|
"history": {
|
|
"recon_loss": [
|
|
0.07256481288653663,
|
|
0.04414264540960137,
|
|
0.03923531813133094,
|
|
0.03690198250512919,
|
|
0.03610035986042558,
|
|
0.03585429300562057,
|
|
0.03614142174936003,
|
|
0.03675156902187528,
|
|
0.03742894164135314,
|
|
0.0383954416307756,
|
|
0.039195723903293796,
|
|
0.04035789030007063,
|
|
0.04136944739705222,
|
|
0.04217637810282982,
|
|
0.04317184983410387,
|
|
0.04414223251529993,
|
|
0.045082664196817286,
|
|
0.0459897641140299,
|
|
0.046940210410672374,
|
|
0.04794481957856661,
|
|
0.04979389414000206,
|
|
0.050820046454731725,
|
|
0.050248638519810304,
|
|
0.049580192527709864,
|
|
0.049034240106359504,
|
|
0.048891564759497456,
|
|
0.04878619169163653,
|
|
0.04837235698524194,
|
|
0.04854527746255581,
|
|
0.04844111150815191,
|
|
0.0483922261872098,
|
|
0.04822369044025739,
|
|
0.048200586078386023,
|
|
0.047983415560144134,
|
|
0.0478166768199995,
|
|
0.0477692344918465,
|
|
0.04770479275661911,
|
|
0.04758516683155655,
|
|
0.047635042673756935,
|
|
0.04756982100761344,
|
|
0.04745731656399802,
|
|
0.04740560844413236,
|
|
0.04746005712793423,
|
|
0.04733826805893172,
|
|
0.047378935079034575,
|
|
0.04739771968024409,
|
|
0.047426708408782624,
|
|
0.047361184023002274,
|
|
0.04733280055861697,
|
|
0.04722336093043415,
|
|
0.04732823220646789,
|
|
0.0472236652022753,
|
|
0.047047707785525895,
|
|
0.04700392111339885,
|
|
0.046977425869713485,
|
|
0.04659851064157282,
|
|
0.04664327050599023,
|
|
0.046512148024625756,
|
|
0.04652243379790049,
|
|
0.04643834348067514,
|
|
0.046477761987246506,
|
|
0.04622124535883339,
|
|
0.04625549639423943,
|
|
0.04613285658992509,
|
|
0.04605264870975262,
|
|
0.04589356125419975,
|
|
0.04583498457462615,
|
|
0.04573038787955148,
|
|
0.0457718974759436,
|
|
0.04564371212329875,
|
|
0.04559122066562756,
|
|
0.045480694916322194,
|
|
0.04536095822755343,
|
|
0.045297518324775576,
|
|
0.04521293488418699,
|
|
0.04522842195076056,
|
|
0.045203558049904995,
|
|
0.04512962291383336,
|
|
0.04501655121516977,
|
|
0.04495001771351975,
|
|
0.044781594123277396,
|
|
0.04481303215854698,
|
|
0.044778383241441004,
|
|
0.044675282958863125,
|
|
0.04455677082395961,
|
|
0.04459468223568466,
|
|
0.04447825120873431,
|
|
0.04439788500372416,
|
|
0.044356526783070505,
|
|
0.04437946719236863,
|
|
0.044278508139790125,
|
|
0.0442026794816439,
|
|
0.04421330326929306,
|
|
0.0441932631099326,
|
|
0.04406989279847879,
|
|
0.043991477594862126,
|
|
0.04401077344440497,
|
|
0.043903381531883,
|
|
0.043854401948360294,
|
|
0.04389296859725673
|
|
],
|
|
"kl_loss": [
|
|
0.9422818552225064,
|
|
0.8304157827654456,
|
|
0.711033024212234,
|
|
0.6201538036012242,
|
|
0.5516812854852432,
|
|
0.49781491747523987,
|
|
0.4539294478475538,
|
|
0.4149613824919758,
|
|
0.3822571930085492,
|
|
0.3525762819708922,
|
|
0.32686203138695824,
|
|
0.3036677601118373,
|
|
0.2831739962228343,
|
|
0.26551324956946903,
|
|
0.24915392639545295,
|
|
0.234747304684586,
|
|
0.22180349417986014,
|
|
0.21073915000654694,
|
|
0.20005385755983174,
|
|
0.19080995443539742,
|
|
0.20008063157144776,
|
|
0.20165701993764976,
|
|
0.20599561308821043,
|
|
0.20952844473286572,
|
|
0.2119306048266908,
|
|
0.21374986164717594,
|
|
0.21439095319081575,
|
|
0.2146782086382055,
|
|
0.21535835013939783,
|
|
0.2156351603822321,
|
|
0.2156747308297035,
|
|
0.2163603354850386,
|
|
0.21604907694153297,
|
|
0.2170932454049078,
|
|
0.2172167382688604,
|
|
0.21753370074125436,
|
|
0.21788769651554588,
|
|
0.21780646761131084,
|
|
0.21757154243114668,
|
|
0.2174266650954373,
|
|
0.21739230457788858,
|
|
0.21768093402059668,
|
|
0.21731426490423006,
|
|
0.21709907592998612,
|
|
0.21729469019123632,
|
|
0.21711179728691393,
|
|
0.2172128687429632,
|
|
0.2170052423309057,
|
|
0.21706495239821255,
|
|
0.2174526669530787,
|
|
0.2170911322737861,
|
|
0.21746987362320608,
|
|
0.21689637371490145,
|
|
0.21768690562910503,
|
|
0.21710836957407814,
|
|
0.21804575870434442,
|
|
0.217907239165571,
|
|
0.21759957396703908,
|
|
0.21808122977232322,
|
|
0.21784972603249753,
|
|
0.21785366121265623,
|
|
0.21845279487534466,
|
|
0.21855741242567697,
|
|
0.21834458735508797,
|
|
0.21865385595677245,
|
|
0.2191770772457632,
|
|
0.21917275520853507,
|
|
0.21897286963131693,
|
|
0.2189757764721528,
|
|
0.21977992066078716,
|
|
0.22024847930058455,
|
|
0.21994173469451758,
|
|
0.21978110724534744,
|
|
0.21947090617484516,
|
|
0.219740368394006,
|
|
0.21953950682256976,
|
|
0.2196803634874841,
|
|
0.2198349738923403,
|
|
0.2205247895903567,
|
|
0.2201039091580444,
|
|
0.22056390408776763,
|
|
0.22043023312575796,
|
|
0.22098283940910274,
|
|
0.22084448502486587,
|
|
0.22079740432847259,
|
|
0.220382661678088,
|
|
0.2211024827745735,
|
|
0.2208555058345326,
|
|
0.22062583204008576,
|
|
0.22087381401250505,
|
|
0.22094633755011436,
|
|
0.22113347610729372,
|
|
0.2210337378594101,
|
|
0.22095281971443412,
|
|
0.22108463678731877,
|
|
0.22107870204963237,
|
|
0.22101640819102272,
|
|
0.221434213889715,
|
|
0.22125727438137063,
|
|
0.22125603885859504
|
|
],
|
|
"perc_loss": [
|
|
3.268518664388575,
|
|
2.9844126731921463,
|
|
2.8893332792143536,
|
|
2.842300501644102,
|
|
2.8177173336346946,
|
|
2.8061808974314957,
|
|
2.80318136449553,
|
|
2.8056567907333374,
|
|
2.8115715369200096,
|
|
2.8198827128125052,
|
|
2.828476501835717,
|
|
2.8390082136178627,
|
|
2.849494279959263,
|
|
2.8572748884176598,
|
|
2.867123058718494,
|
|
2.875808890049274,
|
|
2.8851839202082057,
|
|
2.892486855515048,
|
|
2.9006601964306626,
|
|
2.90893748937509,
|
|
3.002905537430038,
|
|
3.0841518523346663,
|
|
3.1015493930914464,
|
|
3.1056835936684895,
|
|
3.1069560407573342,
|
|
3.1064426521969657,
|
|
3.108147989480923,
|
|
3.1069585961154385,
|
|
3.1126723849875297,
|
|
3.110053312065255,
|
|
3.111884709606823,
|
|
3.109374719297784,
|
|
3.1103647427681165,
|
|
3.1077851240451517,
|
|
3.1043045816258488,
|
|
3.103300364098997,
|
|
3.100696138337127,
|
|
3.101050837427123,
|
|
3.104057640601427,
|
|
3.1031913258071637,
|
|
3.1028205446707897,
|
|
3.0987310552189493,
|
|
3.104932033098661,
|
|
3.100133360451103,
|
|
3.10110838087196,
|
|
3.1032485880403438,
|
|
3.1042901129804106,
|
|
3.1003458077072077,
|
|
3.103027570960868,
|
|
3.0989463599319134,
|
|
3.100031643341749,
|
|
3.0986684830779705,
|
|
3.097568829344888,
|
|
3.0981435597452345,
|
|
3.0955454218081937,
|
|
3.091543544561435,
|
|
3.094627695205884,
|
|
3.0887583132483,
|
|
3.0919363335666494,
|
|
3.0886334053471556,
|
|
3.089639984134935,
|
|
3.0892311589330688,
|
|
3.0903091853500433,
|
|
3.089088388487824,
|
|
3.0876893849454374,
|
|
3.0847945982574396,
|
|
3.083339012076712,
|
|
3.0836611441057973,
|
|
3.0838507724623394,
|
|
3.083322472551949,
|
|
3.0839521288871765,
|
|
3.0830512260779357,
|
|
3.0811995609193787,
|
|
3.077091329627567,
|
|
3.0774103555923853,
|
|
3.0779298793556342,
|
|
3.0798700487511788,
|
|
3.0797718432214527,
|
|
3.0784103956996884,
|
|
3.0775557175660744,
|
|
3.0766860032692933,
|
|
3.0762227615739546,
|
|
3.078130325700483,
|
|
3.0726998621581965,
|
|
3.0729522781494336,
|
|
3.073877077836257,
|
|
3.0732357578399854,
|
|
3.0725482758293805,
|
|
3.0729044237707415,
|
|
3.0753370171938186,
|
|
3.0735755771653266,
|
|
3.073718212099157,
|
|
3.0729570011807303,
|
|
3.073591929725093,
|
|
3.0722544748558955,
|
|
3.0702306806531725,
|
|
3.071104440933619,
|
|
3.070158361369728,
|
|
3.0697627974371624,
|
|
3.0690404686153445
|
|
],
|
|
"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,
|
|
1.1438086142078934,
|
|
1.3848216148714225,
|
|
1.431843824498057,
|
|
1.4550386664107378,
|
|
1.534417847609227,
|
|
1.5428873566456904,
|
|
1.5969856012459351,
|
|
1.6761763638968015,
|
|
1.6849639529728482,
|
|
1.6978890428033013,
|
|
1.7520971780754306,
|
|
1.758636311062763,
|
|
1.7961577434952443,
|
|
1.7791677351849966,
|
|
1.8282766153351364,
|
|
1.8869179757869141,
|
|
1.8936229561073468,
|
|
1.916680078762464,
|
|
2.0205936250396266,
|
|
2.0106747031020813,
|
|
2.1042307398488953,
|
|
2.0886956008198934,
|
|
2.1438577952914133,
|
|
2.144290308058103,
|
|
2.2067003882466216,
|
|
2.260120966622972,
|
|
2.3372040491583,
|
|
2.2787050508981586,
|
|
2.4021470887411356,
|
|
2.406920169790586,
|
|
2.4421350724167294,
|
|
2.4470817596993895,
|
|
2.46695097300232,
|
|
2.470248263870549,
|
|
2.5584849823489146,
|
|
2.492347888998751,
|
|
2.526307321766503,
|
|
2.5345261624226203,
|
|
2.5415179539694743,
|
|
2.553662398814136,
|
|
2.590494660103423,
|
|
2.609607903366415,
|
|
2.5870104153186846,
|
|
2.633848020281547,
|
|
2.617674359271669,
|
|
2.6077225506305695,
|
|
2.626374716687406,
|
|
2.623414141500098,
|
|
2.6401452143222857,
|
|
2.685499318389811,
|
|
2.6790291177920804,
|
|
2.6827956965336432,
|
|
2.6683848851766343,
|
|
2.692997595948032,
|
|
2.7399796782395778,
|
|
2.7001287651877117,
|
|
2.7064600485002894,
|
|
2.7366674155251594,
|
|
2.742031560239629,
|
|
2.7530031089599314,
|
|
2.718017868251882,
|
|
2.733836143444746,
|
|
2.7269677244699917,
|
|
2.7397567153486433,
|
|
2.7358992522598333,
|
|
2.7409965941029735,
|
|
2.74284782852882,
|
|
2.7570678810788016,
|
|
2.744860523276859,
|
|
2.727091629790445,
|
|
2.728150630608583,
|
|
2.7151212447728867,
|
|
2.7098901113893232,
|
|
2.7117845436446686,
|
|
2.6960576665707126,
|
|
2.6918129436990137,
|
|
2.6912999886732836,
|
|
2.7062390074770675,
|
|
2.712936248534765,
|
|
2.713185273683988
|
|
],
|
|
"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.43680395959661555,
|
|
0.34810768227037203,
|
|
0.3377910334075618,
|
|
0.3321381963662103,
|
|
0.2982299141904228,
|
|
0.30104209130836856,
|
|
0.28214932976561224,
|
|
0.26205551579721975,
|
|
0.2678493159608199,
|
|
0.2567906266508194,
|
|
0.25110278342269426,
|
|
0.24189095977598277,
|
|
0.2394267334483373,
|
|
0.24564475484956533,
|
|
0.24081433725216958,
|
|
0.21658801688597754,
|
|
0.213517039664026,
|
|
0.21218282094177526,
|
|
0.18916759920171183,
|
|
0.19284203568170977,
|
|
0.181842301462769,
|
|
0.1791912000785526,
|
|
0.18734478913088384,
|
|
0.17373414262810832,
|
|
0.16635354463424948,
|
|
0.16650411119270656,
|
|
0.15708727274153733,
|
|
0.16674049754237008,
|
|
0.1501258422867355,
|
|
0.15181534979333225,
|
|
0.13657129330672005,
|
|
0.1467462844478014,
|
|
0.1383675114235753,
|
|
0.14689494454516813,
|
|
0.12695099818162048,
|
|
0.12875286992798504,
|
|
0.13186773245469627,
|
|
0.12181158371779144,
|
|
0.12662809188517496,
|
|
0.1175684176114762,
|
|
0.11314824839226073,
|
|
0.11264375143517287,
|
|
0.1165530812393269,
|
|
0.10845673464350084,
|
|
0.10203271553827784,
|
|
0.11023388663306832,
|
|
0.10421399819131336,
|
|
0.0983247641298888,
|
|
0.09406759767817,
|
|
0.09505048260076815,
|
|
0.09383611069021063,
|
|
0.09381456619017145,
|
|
0.08943638993968439,
|
|
0.08595997961556427,
|
|
0.08149021235692641,
|
|
0.08130677379906559,
|
|
0.07870399655025993,
|
|
0.07449058663203484,
|
|
0.07478499058315642,
|
|
0.07468204382154295,
|
|
0.07225174684491423,
|
|
0.07153088875258198,
|
|
0.06767470285328281,
|
|
0.06715900409154785,
|
|
0.06388170873889557,
|
|
0.06149009168227641,
|
|
0.06165273037826658,
|
|
0.05667935721735414,
|
|
0.058510264835487574,
|
|
0.056089043583418444,
|
|
0.05523068375256645,
|
|
0.05533928267705517,
|
|
0.05186875127295717,
|
|
0.05044461999279566,
|
|
0.050165704740052164,
|
|
0.046495266505477265,
|
|
0.04587119478039826,
|
|
0.04412680363051721,
|
|
0.04238344537309156,
|
|
0.041027947104114115
|
|
],
|
|
"fid": {
|
|
"25": 73.48986053466797,
|
|
"50": 51.912357330322266,
|
|
"75": 50.058326721191406,
|
|
"100": 52.463279724121094
|
|
},
|
|
"train_time_s": 2145.0215554237366
|
|
},
|
|
"n_params": 10608451
|
|
} |