Johnny Fernandes
550 lines
14 KiB
JSON
550 lines
14 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,
|
|
"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_3_vae_patchgan",
|
|
"lr": 0.0005,
|
|
"lr_d": 0.0001,
|
|
"beta_kl": 0.005,
|
|
"lambda_perceptual": 0.1,
|
|
"lambda_adversarial": 0.1,
|
|
"ndf_patch": 64
|
|
},
|
|
"history": {
|
|
"recon_loss": [
|
|
0.18861990762699363,
|
|
0.17771862838894892,
|
|
0.17625267265571487,
|
|
0.1655209145357466,
|
|
0.1592490807748758,
|
|
0.1487464247446539,
|
|
0.14603353411150286,
|
|
0.1451508063249863,
|
|
0.1428027419039072,
|
|
0.13992124921682045,
|
|
0.13796657876262808,
|
|
0.13881459523342612,
|
|
0.13775221871323565,
|
|
0.1388729761362585,
|
|
0.1385588059759038,
|
|
0.13607313768922263,
|
|
0.13542115280770847,
|
|
0.1356422903859972,
|
|
0.13568231646512818,
|
|
0.13521548069249362,
|
|
0.13449831399270612,
|
|
0.13600172715373018,
|
|
0.1340917594348773,
|
|
0.1334343557843031,
|
|
0.13268615652480695,
|
|
0.13145199839949098,
|
|
0.13097500251844907,
|
|
0.13025199375155136,
|
|
0.13000157777952331,
|
|
0.13074792513989994,
|
|
0.1292976242227432,
|
|
0.12862894693628335,
|
|
0.1277259295949569,
|
|
0.12738761995146927,
|
|
0.1277741045714953,
|
|
0.12675375163427785,
|
|
0.12609347720176745,
|
|
0.12532664889581183,
|
|
0.12525444449140474,
|
|
0.12526406627944392,
|
|
0.12543218458692232,
|
|
0.12475586244756849,
|
|
0.12460088995723133,
|
|
0.12324439659396298,
|
|
0.12345891071754134,
|
|
0.12453210038634446,
|
|
0.12279523380546488,
|
|
0.12330379579057041,
|
|
0.1228398924582025,
|
|
0.12300096057419084,
|
|
0.12221984697394392,
|
|
0.12254253446928456,
|
|
0.12105376292497684,
|
|
0.12092442976103888,
|
|
0.12121323749231987,
|
|
0.12089209086619891,
|
|
0.12085758982242173,
|
|
0.1203383455833054,
|
|
0.12019133290801293,
|
|
0.11909146809106709,
|
|
0.11888215768859427,
|
|
0.11911316719065365,
|
|
0.11878056021837088,
|
|
0.11874932859443192,
|
|
0.11939801261402093,
|
|
0.1182701296499397,
|
|
0.1178444561540571,
|
|
0.11677229886826797,
|
|
0.11659586265619494,
|
|
0.11801627119127502,
|
|
0.11732396741326039,
|
|
0.11760511196767673,
|
|
0.1185783229322515,
|
|
0.11797440910122843,
|
|
0.11624205051961108,
|
|
0.11628382372805196,
|
|
0.11632549541437218,
|
|
0.11546542154800178,
|
|
0.11584110010383475,
|
|
0.11488010527359115,
|
|
0.1169404620065903,
|
|
0.11736274040980726,
|
|
0.11596904937018696,
|
|
0.11574434848804759,
|
|
0.11690343212750223,
|
|
0.11699441231341444,
|
|
0.11600632573931645,
|
|
0.11448002388525723,
|
|
0.11636082480987932,
|
|
0.1159489756434137,
|
|
0.11456818167024697,
|
|
0.1150473960571819,
|
|
0.11517152621641628,
|
|
0.11601082697255999,
|
|
0.11600699826565564,
|
|
0.11550768732260434,
|
|
0.11478957516324316,
|
|
0.11459140368124358,
|
|
0.11571422099876098,
|
|
0.11537429563000672
|
|
],
|
|
"kl_loss": [
|
|
276.7223569103795,
|
|
120.11881176223103,
|
|
97.074006153987,
|
|
84.7367372594328,
|
|
72.84500447297708,
|
|
58.20365924509163,
|
|
49.934763883933044,
|
|
44.43522434560661,
|
|
41.382670182448166,
|
|
39.62511647053254,
|
|
38.14286702310937,
|
|
37.18682645324968,
|
|
36.580295619801575,
|
|
35.8198753014589,
|
|
35.183456184517624,
|
|
34.65776822505853,
|
|
34.27475022046994,
|
|
33.969623728695076,
|
|
33.58452079642532,
|
|
33.40871410288362,
|
|
33.494045465420456,
|
|
33.57894731586815,
|
|
33.41263916146042,
|
|
33.5053150307419,
|
|
33.61977716592642,
|
|
33.53540447430733,
|
|
33.71047579936492,
|
|
33.75520780351427,
|
|
33.72239826887082,
|
|
33.84841570079836,
|
|
33.89454254737267,
|
|
33.796366430755356,
|
|
33.99421254997579,
|
|
33.82595450246436,
|
|
33.961408395033615,
|
|
33.92151303576608,
|
|
34.04281723804963,
|
|
34.03389532953246,
|
|
34.118032496199646,
|
|
34.17981781103672,
|
|
34.09320658496303,
|
|
34.11978490128477,
|
|
34.08164874508849,
|
|
34.24213214613434,
|
|
34.148709590618424,
|
|
34.237437354193794,
|
|
34.26119823129768,
|
|
34.16253843062963,
|
|
34.523601450471794,
|
|
34.10786265185756,
|
|
34.4834990134606,
|
|
34.436983410109825,
|
|
34.49521413623777,
|
|
34.37277274661594,
|
|
34.40642456315521,
|
|
34.58365509652684,
|
|
34.46540175543891,
|
|
34.62582559667082,
|
|
34.49816356561123,
|
|
34.425088417835724,
|
|
34.689441599397576,
|
|
34.49228691035866,
|
|
34.47755219386174,
|
|
34.66372329353268,
|
|
34.639683275141266,
|
|
34.53895970695039,
|
|
34.615915135440666,
|
|
34.72424408513257,
|
|
34.75890907874474,
|
|
34.440071921063286,
|
|
34.66837335244203,
|
|
34.842535931839905,
|
|
34.497212450728455,
|
|
34.58439721816625,
|
|
34.64464315593752,
|
|
34.63696094252106,
|
|
34.72094595330393,
|
|
34.87894229399852,
|
|
34.89129682687613,
|
|
34.79601346529447,
|
|
35.0156437311417,
|
|
34.86639989543165,
|
|
34.57888742593619,
|
|
34.73608997540596,
|
|
34.919344290708885,
|
|
35.13728781643077,
|
|
34.88758292564979,
|
|
35.08412531502226,
|
|
35.182489802694725,
|
|
35.11944770812988,
|
|
35.09966516902304,
|
|
34.9707060838357,
|
|
35.145755319513825,
|
|
35.09497239039494,
|
|
35.19841175405388,
|
|
35.11572735126202,
|
|
35.285146452422836,
|
|
35.23364441211407,
|
|
35.14119349585639,
|
|
35.07916343721569
|
|
],
|
|
"perc_loss": [
|
|
4.165379504872184,
|
|
4.306908612577324,
|
|
4.406882733361334,
|
|
4.360000091740209,
|
|
4.347245921945979,
|
|
4.322648144685305,
|
|
4.296968904315916,
|
|
4.275595786734524,
|
|
4.264194417713035,
|
|
4.23915512745197,
|
|
4.253835638873597,
|
|
4.237967304184905,
|
|
4.23198981060941,
|
|
4.223654770953024,
|
|
4.2143101213324785,
|
|
4.195945755029336,
|
|
4.195242772245,
|
|
4.195372948279748,
|
|
4.185335707970155,
|
|
4.173882915423467,
|
|
4.16755078236262,
|
|
4.171633275146158,
|
|
4.159881540343293,
|
|
4.167673206227457,
|
|
4.159263728011368,
|
|
4.149037993871248,
|
|
4.139473092861665,
|
|
4.125925103313903,
|
|
4.114537838177803,
|
|
4.116090639024718,
|
|
4.109231275880439,
|
|
4.110566814740499,
|
|
4.096888745951857,
|
|
4.092171801461114,
|
|
4.0924049615859985,
|
|
4.071065393268553,
|
|
4.06966782023764,
|
|
4.061785459008991,
|
|
4.060950119271237,
|
|
4.061661668312856,
|
|
4.056711556565048,
|
|
4.0546094958598795,
|
|
4.048456468133845,
|
|
4.031783304662786,
|
|
4.03786053463944,
|
|
4.048189310946016,
|
|
4.024937762154473,
|
|
4.031513743930393,
|
|
4.041708946737469,
|
|
4.029254699874128,
|
|
4.020909377142915,
|
|
4.022204681848868,
|
|
4.0144958149673595,
|
|
4.014826006868965,
|
|
4.011397586419032,
|
|
4.009836474544982,
|
|
4.000827926346379,
|
|
4.00207864983469,
|
|
3.9983290128218822,
|
|
3.9805039707412067,
|
|
3.985023063981635,
|
|
3.9873007398385267,
|
|
3.9836343282308335,
|
|
3.9778986990961256,
|
|
3.9819785580675826,
|
|
3.963852386189322,
|
|
3.9681489824229836,
|
|
3.9571395709983306,
|
|
3.9520151141362314,
|
|
3.9639965010504437,
|
|
3.962235056436979,
|
|
3.9679050766504727,
|
|
3.970652671960684,
|
|
3.9660502493890943,
|
|
3.9473167010861583,
|
|
3.9518334758587375,
|
|
3.9450023963920073,
|
|
3.939890970531692,
|
|
3.9405837446196466,
|
|
3.9431381770688243,
|
|
3.9528527356620526,
|
|
3.9480060828037753,
|
|
3.933930030745319,
|
|
3.934511334977598,
|
|
3.944071548616784,
|
|
3.932478380508912,
|
|
3.9239180887866225,
|
|
3.922858091501089,
|
|
3.944245843296377,
|
|
3.929251870028993,
|
|
3.9123981141636515,
|
|
3.9128795622760415,
|
|
3.9116759814767756,
|
|
3.9187084161318264,
|
|
3.9145963553689485,
|
|
3.9085161375184345,
|
|
3.915584529057527,
|
|
3.912150815511361,
|
|
3.9175711910948796,
|
|
3.918475313573821
|
|
],
|
|
"adv_g_loss": [
|
|
1.8026715818728902,
|
|
2.23227376344367,
|
|
2.2183715565464435,
|
|
2.1278742490033817,
|
|
2.0971770584583282,
|
|
2.0484086269926696,
|
|
2.03275868207471,
|
|
2.116789224024257,
|
|
2.090583788151415,
|
|
1.9892138968675563,
|
|
1.9692096719677505,
|
|
1.9062933527713275,
|
|
1.894742165073259,
|
|
1.8143773762326145,
|
|
1.8193041879970293,
|
|
1.81429538425281,
|
|
1.8210619932486334,
|
|
1.7941538303191018,
|
|
1.8341339080749701,
|
|
1.8242450552459202,
|
|
1.8905006748600266,
|
|
1.890135900216957,
|
|
1.8945831565313742,
|
|
1.9415043775215108,
|
|
1.9855943465700898,
|
|
2.0438742065447997,
|
|
2.024762265774238,
|
|
2.0435926514144382,
|
|
2.0605728566869455,
|
|
2.137863679956167,
|
|
2.0963172315755205,
|
|
2.141162985625366,
|
|
2.177981933658449,
|
|
2.17414838136134,
|
|
2.2223903614486384,
|
|
2.2023396426413813,
|
|
2.275088946486258,
|
|
2.2592285347736287,
|
|
2.303790853637406,
|
|
2.3635131845839767,
|
|
2.34030539874071,
|
|
2.3638583834354696,
|
|
2.4543145986066923,
|
|
2.3815444054671078,
|
|
2.4706798928956957,
|
|
2.4403553184790487,
|
|
2.4697066351739516,
|
|
2.537486906043994,
|
|
2.5634906901380954,
|
|
2.5490670253833136,
|
|
2.5615056049174223,
|
|
2.6466195793488088,
|
|
2.604182060967144,
|
|
2.657348400889299,
|
|
2.6449959174944806,
|
|
2.656565827309576,
|
|
2.678831669764641,
|
|
2.688901748794776,
|
|
2.68519163182658,
|
|
2.709996796380251,
|
|
2.7448226344636364,
|
|
2.768862698832129,
|
|
2.7961852730084686,
|
|
2.780011566530945,
|
|
2.7335693817108107,
|
|
2.7967236773071127,
|
|
2.7932374021436415,
|
|
2.8125013566424704,
|
|
2.8752479591430764,
|
|
2.884957304622373,
|
|
2.9013289062895327,
|
|
2.8676841870332375,
|
|
2.9157643700257325,
|
|
2.9385765872449956,
|
|
2.971192495435731,
|
|
2.9460317586731706,
|
|
3.006316975905345,
|
|
3.0268147679475637,
|
|
3.0260748012453065,
|
|
3.098839063165534,
|
|
3.1647758919459124,
|
|
3.106013861986307,
|
|
3.110682492072766,
|
|
3.1142931325822816,
|
|
3.1455837691951003,
|
|
3.08899037552695,
|
|
3.1192317396147637,
|
|
3.1001298911551123,
|
|
3.083799856340783,
|
|
3.1157319148381553,
|
|
3.095861378388527,
|
|
3.1165227436611795,
|
|
3.086983766311254,
|
|
3.0861313699657082,
|
|
3.0921488047665,
|
|
3.0940884631923122,
|
|
3.0856645693126907,
|
|
3.0799090566798153,
|
|
3.086693912999243,
|
|
3.093890537054111
|
|
],
|
|
"adv_d_loss": [
|
|
0.3846818933136061,
|
|
0.19303795548840466,
|
|
0.22524197362204137,
|
|
0.23716845485962865,
|
|
0.25527555404756314,
|
|
0.2399256144069199,
|
|
0.25104449234075016,
|
|
0.2408019018988324,
|
|
0.2598570621548555,
|
|
0.2685285050612994,
|
|
0.2780756782899555,
|
|
0.2814109750633311,
|
|
0.30625250518449354,
|
|
0.3203703035146762,
|
|
0.30513450952294546,
|
|
0.30278418525161904,
|
|
0.3132180890593773,
|
|
0.31392792752410614,
|
|
0.2981438948653447,
|
|
0.3000434341434485,
|
|
0.2830737477693802,
|
|
0.2774529387521693,
|
|
0.27726965177899754,
|
|
0.2697514830969083,
|
|
0.2631684033056864,
|
|
0.2501653985430797,
|
|
0.2529780987777516,
|
|
0.24266270441440946,
|
|
0.25031912855358207,
|
|
0.22288694602047276,
|
|
0.2343698614077945,
|
|
0.2288323061174562,
|
|
0.22137120039735594,
|
|
0.22300533291239005,
|
|
0.20846376789367607,
|
|
0.21844437852119789,
|
|
0.19620536157909113,
|
|
0.20161379222622794,
|
|
0.19350602869422007,
|
|
0.1940044454561594,
|
|
0.19748605735217914,
|
|
0.18612161299420726,
|
|
0.18449116273759267,
|
|
0.1786681996164923,
|
|
0.17217153660220724,
|
|
0.1815700484917332,
|
|
0.17118102850981504,
|
|
0.1629717636646496,
|
|
0.1747021592166434,
|
|
0.1594025977433492,
|
|
0.1572152761519592,
|
|
0.1572253421649464,
|
|
0.14647067149575704,
|
|
0.14247114893494764,
|
|
0.14077772017027068,
|
|
0.1379101731392563,
|
|
0.13861483306042913,
|
|
0.13286431546872243,
|
|
0.1263120919545619,
|
|
0.12339711796619698,
|
|
0.12401506412997206,
|
|
0.11199609567530644,
|
|
0.11085018765929545,
|
|
0.11338936790632896,
|
|
0.12032712383084318,
|
|
0.10354128184640764,
|
|
0.1036931567817417,
|
|
0.10232575738436392,
|
|
0.08907810818308438,
|
|
0.08665908210409376,
|
|
0.09036243790521836,
|
|
0.09317310632835342,
|
|
0.08505074158868092,
|
|
0.08004704910593155,
|
|
0.07433714670264441,
|
|
0.0742792083332554,
|
|
0.07248415664618468,
|
|
0.06889059219676523,
|
|
0.06714722003915116,
|
|
0.05902106333205588,
|
|
0.05658946855940944,
|
|
0.059529210504494674,
|
|
0.05614422008785236,
|
|
0.05332565255678044,
|
|
0.0521102087576197,
|
|
0.052153651256305285,
|
|
0.04740947977098453,
|
|
0.048982632026060395,
|
|
0.05126904483494532,
|
|
0.046512605652260855,
|
|
0.04599587608757628,
|
|
0.04359768634327711,
|
|
0.0426614616647299,
|
|
0.04159874533120002,
|
|
0.03934315871845326,
|
|
0.03652780742233253,
|
|
0.036582552890579864,
|
|
0.03528311501384482,
|
|
0.034168972188017815,
|
|
0.03123981755378091
|
|
],
|
|
"fid": {
|
|
"25": 217.87948608398438,
|
|
"50": 231.8101043701172,
|
|
"75": 233.5815887451172,
|
|
"100": 234.91502380371094
|
|
},
|
|
"train_time_s": 2140.041033744812
|
|
},
|
|
"n_params": 10608451
|
|
} |