546 lines
7.6 KiB
JSON
546 lines
7.6 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,
|
|
"run_name": "p3_2_vae_perceptual",
|
|
"lr": 0.001,
|
|
"beta_kl": 0.0001,
|
|
"lambda_perceptual": 0.1,
|
|
"lambda_adversarial": 0.0
|
|
},
|
|
"history": {
|
|
"recon_loss": [
|
|
NaN,
|
|
0.041370747770127066,
|
|
0.035506031866002284,
|
|
0.06229733923672993,
|
|
0.06089382184048494,
|
|
0.053464704654856116,
|
|
0.04950355895213846,
|
|
0.04674786329269409,
|
|
0.05693557829811023,
|
|
0.04808994451076047,
|
|
NaN,
|
|
0.05869839608701121,
|
|
0.05293231666820426,
|
|
0.05057006603918779,
|
|
0.04713928615117175,
|
|
0.04481589820427008,
|
|
0.04331832689543565,
|
|
0.041820655449524395,
|
|
NaN,
|
|
0.24315394992884407,
|
|
0.24305414841470555,
|
|
0.2429187158998261,
|
|
0.24289727962424612,
|
|
0.2377373814328104,
|
|
0.23880945107875726,
|
|
0.24047312904626894,
|
|
0.23915709144411942,
|
|
NaN,
|
|
0.24000247061634675,
|
|
0.2362435321904655,
|
|
0.23705266075383905,
|
|
NaN,
|
|
0.23998981039238793,
|
|
0.23828768376738596,
|
|
0.23908851302077627,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN
|
|
],
|
|
"kl_loss": [
|
|
41526536.5961082,
|
|
1258.7768262553418,
|
|
1165.5987154968784,
|
|
3689.321748260759,
|
|
1320.3167513333833,
|
|
928.3473894168169,
|
|
789.754654843583,
|
|
693.4729607736963,
|
|
1017.8856519389357,
|
|
622.2623097998464,
|
|
304942.2120989938,
|
|
750.5718396830763,
|
|
618.6888906364767,
|
|
586.4194498958751,
|
|
579.2793892102363,
|
|
525.8830437945504,
|
|
497.5363791050055,
|
|
452.93342662061383,
|
|
172476214612.95245,
|
|
1112.027792645316,
|
|
1109.8156276605068,
|
|
1109.8872808472722,
|
|
1112.3999198196282,
|
|
1163.6093267457097,
|
|
1195.4633280436199,
|
|
1232.709252512353,
|
|
1253.0651078183428,
|
|
19923.65347159622,
|
|
6871.588723207132,
|
|
6872.017739842081,
|
|
6872.62934758113,
|
|
27062.17111336472,
|
|
4992.576698759683,
|
|
5000.895510942508,
|
|
5011.723715236044,
|
|
4.196954763212122e+20,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN
|
|
],
|
|
"perc_loss": [
|
|
NaN,
|
|
3.0017659954535656,
|
|
2.8931196978968434,
|
|
3.084328340159522,
|
|
3.1398269641093717,
|
|
3.0839009172896032,
|
|
3.0459561949102287,
|
|
3.015714469628456,
|
|
3.0917934143645134,
|
|
3.010985380054539,
|
|
NaN,
|
|
3.102917709411719,
|
|
3.055045795746339,
|
|
3.0351512865123587,
|
|
3.002274121993627,
|
|
2.9754957128793764,
|
|
2.9553630066733074,
|
|
2.936704080328982,
|
|
NaN,
|
|
3.66463458385223,
|
|
3.6419444985878773,
|
|
3.6321474706005845,
|
|
3.6272282503608966,
|
|
3.781295938878997,
|
|
3.740793755421272,
|
|
3.759704489993234,
|
|
3.7078149914741516,
|
|
NaN,
|
|
3.7647176323792872,
|
|
3.7441278461717133,
|
|
3.7264530261357627,
|
|
NaN,
|
|
3.7299226170931106,
|
|
3.7314435080585318,
|
|
3.754668933713538,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN,
|
|
NaN
|
|
],
|
|
"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": 263.1458740234375,
|
|
"50": 598.3736572265625,
|
|
"75": 598.3736572265625,
|
|
"100": 598.3736572265625
|
|
},
|
|
"train_time_s": 952.6596128940582
|
|
},
|
|
"n_params": 10608451
|
|
} |