jalf/DRL_PROJ
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Testing VAE until it works - v1

Browse Source
This commit is contained in:
Johnny Fernandes
2026-05-02 13:11:56 +01:00
parent c7804d2984
commit ec8d4ae336
84 changed files with 9 additions and 1744 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/outputs/logs/p3_1_vae.json
-548
View File
@@ -1,548 +0,0 @@
{
"run_name": "p3_1_vae",
"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_1_vae",
"lr": 0.0005,
"beta_kl": 0.5,
"lambda_perceptual": 0.0,
"lambda_adversarial": 0.0
},
"history": {
"recon_loss": [
0.16871115132274792,
0.16150351059742463,
0.15984480073436713,
0.15590801271490562,
0.15061364529861343,
0.14644645811973983,
0.14328488393917552,
0.14094583944887176,
0.13928856181665364,
0.1377539750124909,
0.13583827684195632,
0.13372899556898662,
0.13197963861509776,
0.1295166944559568,
0.12748059913770765,
0.126394641561768,
0.12385777387226748,
0.12244280847983482,
0.121336308045265,
0.12060208647296979,
0.11928399897411339,
0.11839368250061813,
0.11770952168183449,
0.11709171382344177,
0.11688287127922234,
0.11606283619617805,
0.11564522911595483,
0.11527438517500702,
0.11461337662150717,
0.11428412470297936,
0.11390002192849787,
0.11359650807248221,
0.11298466332129434,
0.11281277696227926,
0.11266172842846976,
0.11232183446996233,
0.11213540144137338,
0.11184148606645246,
0.11143260068682015,
0.11121100015365161,
0.11103646766044135,
0.11086450471009454,
0.11068084617901562,
0.11046731698080006,
0.1101092367600172,
0.11040792095228139,
0.11019570772082378,
0.10987201248669726,
0.10948091489063878,
0.10964858110070738,
0.10938817380457862,
0.10904825658688688,
0.10926413088718541,
0.10854257479246356,
0.10841736453784327,
0.10838394499041586,
0.10795553592153084,
0.10781666870491627,
0.10770979084265538,
0.1076134722520653,
0.10732251404123938,
0.10716411035157676,
0.10698133315413426,
0.10672279318364766,
0.10645651288776316,
0.10662059826601265,
0.10623115535156849,
0.10627176197102436,
0.1060571956456217,
0.10596368315382901,
0.10585042324840513,
0.1054028309085685,
0.10517520199601467,
0.1053342710639167,
0.1052525288815427,
0.10502007727821668,
0.10490142727573203,
0.10471088643002714,
0.10447397850390173,
0.1044878327470814,
0.10447094976328887,
0.10421697295501701,
0.10414895819675209,
0.10414911504102568,
0.10366271499894623,
0.10378186422217096,
0.10345360714719336,
0.10363478323396964,
0.10342726219668348,
0.10305201177859408,
0.10310276317545491,
0.10302646558445233,
0.10291730190635237,
0.10278329038276122,
0.10258485910156344,
0.10270861135079311,
0.10251107273830308,
0.10234160087684281,
0.10214911544552216,
0.10251628613879538
],
"kl_loss": [
29.481668366326225,
26.226897622784996,
25.98807217932155,
25.807672516912476,
25.701779809772457,
25.66069327052842,
25.639777212061432,
25.62703147301307,
25.620143454299015,
25.61577053966685,
25.612662934849403,
25.60999612319164,
25.608206019442306,
25.60651744940342,
25.605323510292248,
25.60462644772652,
25.603682958162747,
25.603215930808304,
25.602825727218235,
25.602524419116158,
25.602402703374878,
25.602261029756985,
25.602163278139553,
25.602091275728664,
25.602065310518967,
25.602008819580078,
25.601970770420174,
25.601926473470833,
25.601869432335224,
25.60185879519862,
25.601852824545315,
25.60183112234132,
25.601783161489372,
25.601782627594776,
25.601758304824177,
25.601712039393238,
25.601732046176227,
25.601690267905212,
25.60164840404804,
25.601653335440872,
25.60167118626782,
25.601645954653748,
25.60163675210415,
25.601624191316784,
25.601573104532356,
25.60163724524343,
25.601609873975445,
25.601576218238243,
25.601538580706997,
25.601551618331516,
25.60154583922818,
25.6015028138446,
25.60153263858241,
25.601467629783173,
25.60146150018415,
25.601416970929527,
25.601392692989773,
25.601404438670883,
25.601370538401806,
25.60135773308257,
25.601353584191738,
25.60130478785588,
25.601299799405613,
25.60127318618644,
25.6012494054615,
25.601233796176746,
25.601193383208706,
25.601183601933666,
25.60118840698503,
25.601158150240906,
25.60114165656587,
25.601124979491928,
25.601091221866444,
25.60108805925418,
25.601056249732647,
25.601044960511036,
25.601008244049854,
25.600999204521504,
25.60097885539389,
25.60095446741479,
25.600959500695904,
25.600936054164528,
25.60091640195276,
25.600889332274086,
25.600879269787388,
25.600861647190193,
25.600824853293915,
25.6008102914207,
25.600796605786705,
25.600769034817688,
25.600744826161964,
25.600724538167317,
25.60070591298943,
25.600691750518276,
25.600666380336143,
25.60064803636991,
25.60063368642432,
25.600601587540066,
25.60058915309417,
25.600568995516525
],
"perc_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_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": 238.42819213867188,
"50": 232.70050048828125,
"75": 234.88893127441406,
"100": 236.51181030273438
},
"train_time_s": 676.644668340683
},
"n_params": 10608451
}