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,
|
|
"epochs": 100,
|
|
"augment": "hflip",
|
|
"image_size": 64,
|
|
"model": "vae",
|
|
"latent_dim": 256,
|
|
"ngf": 64,
|
|
"free_bits": 0.1,
|
|
"grad_clip": 1.0,
|
|
"run_name": "p3_2_vae_perceptual",
|
|
"lr": 0.0005,
|
|
"beta_kl": 0.005,
|
|
"lambda_perceptual": 0.1,
|
|
"lambda_adversarial": 0.0
|
|
},
|
|
"history": {
|
|
"recon_loss": [
|
|
0.07706982396447505,
|
|
0.06330727752393638,
|
|
0.060562346210209735,
|
|
0.06067247126792741,
|
|
0.06168853705669315,
|
|
0.06303556570703657,
|
|
0.06482723723842293,
|
|
0.06669721933097666,
|
|
0.0683129624200937,
|
|
0.06987012332926194,
|
|
0.07130966042606239,
|
|
0.07270127332681774,
|
|
0.07425614913654888,
|
|
0.07580115537867586,
|
|
0.07719200621876451,
|
|
0.078589656127569,
|
|
0.07977109738331065,
|
|
0.08097509191268021,
|
|
0.0821382127638556,
|
|
0.08324004981953365,
|
|
0.08320301078642026,
|
|
0.08301973822088833,
|
|
0.08318755321968825,
|
|
0.08317961664790781,
|
|
0.08314789212348624,
|
|
0.08329373801875319,
|
|
0.08338832993728992,
|
|
0.08336645318593225,
|
|
0.08318959130372247,
|
|
0.08320376128913501,
|
|
0.08317808063430154,
|
|
0.08312527894273274,
|
|
0.08306075425611602,
|
|
0.08289793944065897,
|
|
0.0828752614175662,
|
|
0.08287284457976492,
|
|
0.08268047039771181,
|
|
0.08271352856014019,
|
|
0.0829078034362477,
|
|
0.0827700739614984,
|
|
0.0827989391823355,
|
|
0.08292064927199967,
|
|
0.0828404097029796,
|
|
0.08288907250150657,
|
|
0.0829680990268532,
|
|
0.08282275031456071,
|
|
0.08290983905267511,
|
|
0.08281088118942884,
|
|
0.08267007887554474,
|
|
0.08269618300354889,
|
|
0.0827041374057786,
|
|
0.0825083223131732,
|
|
0.08238164631602092,
|
|
0.08237539884498996,
|
|
0.0822869433870173,
|
|
0.08221659919199271,
|
|
0.08213427105647886,
|
|
0.08218342011683008,
|
|
0.082064455613876,
|
|
0.08187834614426152,
|
|
0.08198784355424409,
|
|
0.08176947068263832,
|
|
0.08174051175642218,
|
|
0.08147429703519894,
|
|
0.08165825637550946,
|
|
0.08149343700363086,
|
|
0.08137227281227581,
|
|
0.08143208594594756,
|
|
0.08119541610408033,
|
|
0.08130555122326581,
|
|
0.08124906267238478,
|
|
0.08108980721260747,
|
|
0.08119610937423685,
|
|
0.08095862009586433,
|
|
0.08084884440351246,
|
|
0.08087979268250811,
|
|
0.08086979661423427,
|
|
0.08077665719275291,
|
|
0.08070505911914203,
|
|
0.08077280767835103,
|
|
0.0806472651564922,
|
|
0.08055042177757137,
|
|
0.08046174721203299,
|
|
0.08041473021173579,
|
|
0.08053297676846512,
|
|
0.08046999694699915,
|
|
0.0805438561603809,
|
|
0.0802764052636603,
|
|
0.08026964488065141,
|
|
0.08028597289170974,
|
|
0.0801090358191321,
|
|
0.08017482723181064,
|
|
0.08005877630387107,
|
|
0.08010221279075003,
|
|
0.07999415299258171,
|
|
0.0798476467975694,
|
|
0.07997323754124153,
|
|
0.0798028621217634,
|
|
0.07988279134544551,
|
|
0.07989695622052392
|
|
],
|
|
"kl_loss": [
|
|
114.89598556257721,
|
|
77.33424778473683,
|
|
62.9100211705917,
|
|
55.875159540746964,
|
|
51.501892236562874,
|
|
48.34236133608044,
|
|
45.885433547517174,
|
|
43.94423294067383,
|
|
42.245688601436775,
|
|
40.89320367829413,
|
|
39.69322418147682,
|
|
38.6090751876179,
|
|
37.7343929486397,
|
|
36.89236228486411,
|
|
36.14446016458365,
|
|
35.46583691621438,
|
|
34.81329877967509,
|
|
34.25002269255809,
|
|
33.78917465862046,
|
|
33.34034808069213,
|
|
33.22343346196362,
|
|
33.14858524615948,
|
|
33.055273805928024,
|
|
32.99898836347792,
|
|
32.90522269713573,
|
|
32.836681887634796,
|
|
32.76146248874501,
|
|
32.68683499556322,
|
|
32.632087422232345,
|
|
32.59484779325306,
|
|
32.54658634642251,
|
|
32.49264952260205,
|
|
32.471018106509476,
|
|
32.43708103946132,
|
|
32.433846559280006,
|
|
32.40765173822387,
|
|
32.363136197766686,
|
|
32.320940767597946,
|
|
32.29645855813964,
|
|
32.26480372746786,
|
|
32.22680737829616,
|
|
32.176548860011955,
|
|
32.135557647444244,
|
|
32.11595243877835,
|
|
32.05515718867636,
|
|
32.04982715997941,
|
|
32.00999742084079,
|
|
31.98886978931916,
|
|
31.965849428095368,
|
|
31.92324785085825,
|
|
31.916118495484703,
|
|
31.915471476367397,
|
|
31.903850473909298,
|
|
31.890775949527058,
|
|
31.8645691993909,
|
|
31.869942917783035,
|
|
31.860487742301746,
|
|
31.850944148169624,
|
|
31.843142101907322,
|
|
31.839001268403145,
|
|
31.847271275316547,
|
|
31.829896576384193,
|
|
31.801259293515457,
|
|
31.81849612130059,
|
|
31.774390408116528,
|
|
31.791951089842705,
|
|
31.79695913526747,
|
|
31.78303485038953,
|
|
31.772433419512886,
|
|
31.75331918602316,
|
|
31.766783274137058,
|
|
31.768596893701798,
|
|
31.765920822436993,
|
|
31.724387494926777,
|
|
31.745759788741413,
|
|
31.730095512846596,
|
|
31.737777669205624,
|
|
31.726243728246445,
|
|
31.71437902532072,
|
|
31.71155218792777,
|
|
31.704629376403286,
|
|
31.707098475888245,
|
|
31.703360154078556,
|
|
31.695194660088955,
|
|
31.685226285559498,
|
|
31.6823458956857,
|
|
31.674420800983395,
|
|
31.686162447318054,
|
|
31.676387077722794,
|
|
31.680302566952175,
|
|
31.657646749773598,
|
|
31.67580248759343,
|
|
31.65489924259675,
|
|
31.662759666768913,
|
|
31.655795794266922,
|
|
31.663332922845825,
|
|
31.652285494356075,
|
|
31.666093015263222,
|
|
31.64289121546297,
|
|
31.66365980083107
|
|
],
|
|
"perc_loss": [
|
|
3.347367505232493,
|
|
3.1841943034758935,
|
|
3.1360840721008105,
|
|
3.1123677431008754,
|
|
3.102960804079333,
|
|
3.0996367656267605,
|
|
3.100349969843514,
|
|
3.1043578926314654,
|
|
3.1047080194848213,
|
|
3.10818816963424,
|
|
3.1107087919854712,
|
|
3.11433150116195,
|
|
3.1165808979262652,
|
|
3.1217524506087995,
|
|
3.124025124769944,
|
|
3.1286499158948913,
|
|
3.1313081755597367,
|
|
3.135032072535947,
|
|
3.1376487164415865,
|
|
3.140188893191835,
|
|
3.138313090699351,
|
|
3.136352961898869,
|
|
3.1350333721209793,
|
|
3.1340697214134736,
|
|
3.1320774601056027,
|
|
3.1314907603793674,
|
|
3.1301532255278692,
|
|
3.12954652411306,
|
|
3.12765454991251,
|
|
3.126848427658407,
|
|
3.1259387716268883,
|
|
3.1238634581239815,
|
|
3.1228250631919274,
|
|
3.1218308898118825,
|
|
3.120684378167503,
|
|
3.1207721839603195,
|
|
3.1186747016050878,
|
|
3.118101167373168,
|
|
3.11715858308678,
|
|
3.1159399615393744,
|
|
3.1157061961980967,
|
|
3.115308819163559,
|
|
3.114226241906484,
|
|
3.113662652989738,
|
|
3.112562283491477,
|
|
3.1123410055780005,
|
|
3.1108450038820252,
|
|
3.110834915413816,
|
|
3.11004300453724,
|
|
3.1096309413257828,
|
|
3.1089638593869333,
|
|
3.1076014179449816,
|
|
3.1071455493951454,
|
|
3.1054801660725193,
|
|
3.1046676870085235,
|
|
3.1034356477933054,
|
|
3.1028999105477943,
|
|
3.10203901773844,
|
|
3.100277382084447,
|
|
3.0997838067193317,
|
|
3.0997486939797034,
|
|
3.0976629460978713,
|
|
3.096854481432173,
|
|
3.0961881641648774,
|
|
3.0957974134347377,
|
|
3.094707619940114,
|
|
3.0938115603903418,
|
|
3.0930811013930883,
|
|
3.092930595589499,
|
|
3.091999230731247,
|
|
3.0908807531381264,
|
|
3.0901810919117723,
|
|
3.089734352551974,
|
|
3.0887419851417217,
|
|
3.087375036162189,
|
|
3.088294555998256,
|
|
3.087375529301472,
|
|
3.0865699587724147,
|
|
3.085814117366432,
|
|
3.086571911970774,
|
|
3.0848990953885593,
|
|
3.0839059826655264,
|
|
3.0831500668810983,
|
|
3.082290298408932,
|
|
3.083427106213366,
|
|
3.0819556769142804,
|
|
3.0830693351916776,
|
|
3.080666415202312,
|
|
3.0801111816341042,
|
|
3.080527792119572,
|
|
3.0795028067042685,
|
|
3.078920972143483,
|
|
3.078586568689754,
|
|
3.0789812258166127,
|
|
3.0772171998635316,
|
|
3.077096633422069,
|
|
3.0768553665560536,
|
|
3.0771814630581784,
|
|
3.076374116616371,
|
|
3.076963826122447
|
|
],
|
|
"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": 244.7733612060547,
|
|
"50": 231.2705535888672,
|
|
"75": 229.399658203125,
|
|
"100": 230.25457763671875
|
|
},
|
|
"train_time_s": 1399.3615255355835
|
|
},
|
|
"n_params": 10608451
|
|
} |