diff --git a/generator/outputs/logs/p5_vae.json b/generator/outputs/logs/p5_vae.json new file mode 100644 index 0000000..49fa44d --- /dev/null +++ b/generator/outputs/logs/p5_vae.json @@ -0,0 +1,551 @@ +{ + "run_name": "p5_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, + "num_workers": 2, + "run_name": "p5_vae", + "model": "vae", + "epochs": 400, + "augment": "hflip", + "image_size": 64, + "latent_dim": 256, + "ngf": 64, + "lr": 0.001, + "lr_d": 0.0001, + "beta_kl": 0.25, + "lambda_perceptual": 0.1, + "lambda_adversarial": 0.1, + "ndf_patch": 64, + "max_train_hours": 24, + "fid_patience": 2 + }, + "history": { + "recon_loss": [ + 0.0873407603551944, + 0.04358899736235666, + 0.03418333942675565, + 0.029945257139734477, + 0.02730485076148413, + 0.0256642885824554, + 0.024606309914722655, + 0.024003406015471514, + 0.023481889054752313, + 0.023388028817458287, + 0.023165030958146877, + 0.023034087935478516, + 0.023093829576212626, + 0.023127540110204466, + 0.023243789039711412, + 0.023344778173213076, + 0.023640765824442744, + 0.023751988847795714, + 0.02397893554268357, + 0.024262953263062697, + 0.024448917048354436, + 0.024872611857887007, + 0.02515550658234164, + 0.02550989743242534, + 0.025736652398243163, + 0.02602377253321883, + 0.02638998716814905, + 0.026764424573470894, + 0.02702379642197719, + 0.027440408292489175, + 0.027744242869731452, + 0.028089644061003484, + 0.02848421647730801, + 0.028713980769245032, + 0.029125269327280868, + 0.029430037913605187, + 0.029704269650591236, + 0.03008795604826166, + 0.030501786245303787, + 0.030806709439135514, + 0.03112352060743122, + 0.03151081111998512, + 0.03180294511760147, + 0.03201285475849086, + 0.03250477283110476, + 0.03279393227197803, + 0.033106526129075095, + 0.03341933650673073, + 0.03377721751204286, + 0.03395956884432807, + 0.034379123602635585, + 0.03460381771477624, + 0.03494038589250965, + 0.03528320585958596, + 0.035614780918496035, + 0.035869904081376955, + 0.03618363298985184, + 0.036493296719068646, + 0.03672212112344738, + 0.0370304095121021, + 0.037354495376348495, + 0.03760785502023422, + 0.037863354250374764, + 0.03823175100195739, + 0.038425643482587785, + 0.03870724631139101, + 0.0389937731063264, + 0.03924838143090407, + 0.03949139775055596, + 0.03976541371522551, + 0.03993153934263521, + 0.040273130831555426, + 0.04053226028943164, + 0.04074429119658521, + 0.041034609883323185, + 0.04132901987809147, + 0.04159073577795783, + 0.04180888088945395, + 0.04201203822675678, + 0.042261788796665326, + 0.08396607959786287, + 0.10181164084018296, + 0.09799152593582104, + 0.09668531939068921, + 0.0968421785813621, + 0.09369673845795994, + 0.09009833600467597, + 0.0903104149346423, + 0.08894572399047196, + 0.08763274517005835, + 0.08656074902695468, + 0.08540202769586164, + 0.08515783267207125, + 0.08472549243487863, + 0.08296199838638815, + 0.08328578632293093, + 0.08295941359212256, + 0.08313999250212796, + 0.0818887078959463, + 0.08250387434839693 + ], + "kl_loss": [ + 1.8525980165880969, + 1.4553406062798622, + 1.3945304973003192, + 1.3407280745669308, + 1.2857661698109064, + 1.2336315864171736, + 1.1874862180815802, + 1.1414925043399518, + 1.0977480350396571, + 1.0560076374274034, + 1.01724683932769, + 0.9797984164240013, + 0.9446675681915039, + 0.9104400175249475, + 0.8793187248401153, + 0.8478198795237093, + 0.8173006469877357, + 0.7886208149357739, + 0.7629873515703739, + 0.7364261971834378, + 0.7128106751757809, + 0.6888662757527115, + 0.6657254862734395, + 0.6452364749633349, + 0.6245541180300916, + 0.6049946047301985, + 0.5857444695937328, + 0.5684402824467064, + 0.5521934639949065, + 0.5358257534412237, + 0.5204513295211344, + 0.5060414804989456, + 0.49239281348438346, + 0.4787919474208457, + 0.4658316922111389, + 0.4539918888073701, + 0.4419420047575592, + 0.4312195180573015, + 0.42050094068305105, + 0.40945692296720976, + 0.40024506561776513, + 0.39102863456703657, + 0.3820533954944366, + 0.3735176674448527, + 0.3652671699722608, + 0.35739283327363497, + 0.3497536080515283, + 0.3421799224666041, + 0.3350429548284946, + 0.32887285833175367, + 0.3223056475448812, + 0.31573223348102, + 0.30969550721665734, + 0.3040077604162387, + 0.2983994829094308, + 0.2931375425213423, + 0.28794814818180525, + 0.28312774658457845, + 0.27795972312108064, + 0.2733073151137075, + 0.26867456909300935, + 0.2644821102611529, + 0.26038518162746715, + 0.2562667843368318, + 0.25174618904025126, + 0.24832279088660184, + 0.24432727156413925, + 0.24079690383285537, + 0.2376329786120317, + 0.2344166113334334, + 0.23053084180141106, + 0.22791345137306768, + 0.22474394993394867, + 0.22178132660113847, + 0.2186352158140423, + 0.21556782238503808, + 0.21307267468327132, + 0.21009610163477752, + 0.20776716148496693, + 0.2049679136556438, + 0.2247045011792937, + 0.2217696603292074, + 0.22496450988528055, + 0.22926593384044802, + 0.22361361193988058, + 0.22135189595894936, + 0.22035350220707747, + 0.22777206256476223, + 0.22325334799850088, + 0.2233544290702567, + 0.22975832863877982, + 0.22763504036980817, + 0.2286809598788237, + 0.23215497132295218, + 0.230580695769471, + 0.2321891579299401, + 0.23496549621097043, + 0.23647785600688723, + 0.23949947577511144, + 0.24036300223734644 + ], + "perc_loss": [ + 3.335951432203635, + 2.998757926317362, + 2.8297285636266074, + 2.727970255745782, + 2.6565276032839065, + 2.6064420164140882, + 2.5682258865772147, + 2.5430078180427227, + 2.522596321044824, + 2.5114925208254757, + 2.5005857857883487, + 2.49486218405585, + 2.491978524077652, + 2.490892021574526, + 2.4906904473263993, + 2.492909249586937, + 2.497568679161561, + 2.5003321379677863, + 2.505250292456048, + 2.5104152916843057, + 2.516255518819532, + 2.523510086740184, + 2.53035162605791, + 2.536790308279869, + 2.542752216514359, + 2.55020610949932, + 2.5582634582478776, + 2.565885178044311, + 2.5708479911853104, + 2.5798426958230825, + 2.5859895343454475, + 2.5931876342520757, + 2.60077048621626, + 2.6064286018029237, + 2.6146339947341852, + 2.620219861340319, + 2.6261617546407585, + 2.63408244929762, + 2.6406683646715603, + 2.6461050449273524, + 2.6516372836553135, + 2.6584073289846764, + 2.6636348529758616, + 2.6680258059094095, + 2.6752815042805467, + 2.680128652315873, + 2.6852204356438074, + 2.69005358677644, + 2.6959197098373346, + 2.6993423679954986, + 2.7052695216276708, + 2.709015153412126, + 2.7140026301400275, + 2.7192573567740936, + 2.7227722610163894, + 2.727261029756986, + 2.731931083732181, + 2.735040227572123, + 2.739379578675979, + 2.7434892078750153, + 2.747702214962397, + 2.7503509246386013, + 2.7542025533496823, + 2.7583101478397336, + 2.7612236325557413, + 2.76464156819205, + 2.768620821655306, + 2.7716130385032067, + 2.7744738800912843, + 2.7773860871282396, + 2.780290570014562, + 2.784183261740921, + 2.786414759281354, + 2.78930699876231, + 2.7921430218932977, + 2.7955944777553916, + 2.7981467792111583, + 2.800831956231696, + 2.8030711285069456, + 2.8052900795243745, + 3.519835437974359, + 3.8608342984802704, + 3.8944253116591363, + 3.943398314154046, + 3.962524364137242, + 3.9387727614141936, + 3.9044231028638334, + 3.912559205140823, + 3.9143120524210806, + 3.898512967631348, + 3.8972727090884476, + 3.8735957543055215, + 3.872446519187373, + 3.863863379527361, + 3.8498869344719453, + 3.841842463383308, + 3.847525435635167, + 3.841787440654559, + 3.814141992829804, + 3.8276908326352763 + ], + "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.5217359245707018, + 0.981920200933376, + 1.2374933425846517, + 1.2787476180710344, + 1.3618476113511457, + 1.2804014555409424, + 1.245144590241914, + 1.3317598049711978, + 1.3350145901567263, + 1.3249803283212023, + 1.4742424943082384, + 1.4282420477232871, + 1.485014318018897, + 1.5059755784976814, + 1.5204202550001698, + 1.5489312943059186, + 1.6689910165503876, + 1.6956130060721508, + 1.681979666503831, + 1.782702047811646 + ], + "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.8759659508354644, + 0.6683682966181356, + 0.5347345314092107, + 0.5164251379732393, + 0.4872464276372622, + 0.49361801707846487, + 0.503935003382528, + 0.47848207029139894, + 0.4685567660718902, + 0.4727442391917237, + 0.42426876004180336, + 0.4130457585247663, + 0.40487297226348495, + 0.38646075183637124, + 0.39057583349128056, + 0.37396795503222025, + 0.33111938719566053, + 0.34095216335521805, + 0.3216267083254125, + 0.3050928452076056 + ], + "fid": { + "25": 86.1589584350586, + "50": 56.15982437133789, + "75": 60.979278564453125, + "100": 58.461761474609375 + }, + "train_time_s": 1603.9792292118073 + }, + "n_params": 10608451 +} \ No newline at end of file diff --git a/generator/outputs/models/p5_vae_best_ema.pt b/generator/outputs/models/p5_vae_best_ema.pt new file mode 100644 index 0000000..fb8a253 --- /dev/null +++ b/generator/outputs/models/p5_vae_best_ema.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f74239e90081612ef19d9ae1393896ff3db774c0bbb85ff6cdf0e5144dbc7f0 +size 42443581 diff --git a/generator/outputs/models/p5_vae_best_vae.pt b/generator/outputs/models/p5_vae_best_vae.pt new file mode 100644 index 0000000..ef81f81 --- /dev/null +++ b/generator/outputs/models/p5_vae_best_vae.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3cc5df8130098ba9b3bd903d589f2042db87570c138bfeb7bc92cc41a6228f83 +size 42444221 diff --git a/generator/outputs/models/p5_vae_final_ema.pt b/generator/outputs/models/p5_vae_final_ema.pt new file mode 100644 index 0000000..4f4dad2 --- /dev/null +++ b/generator/outputs/models/p5_vae_final_ema.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d69d509722767ff1e1c1849454ef9566c33a7015187605c244754bae86e76c40 +size 42443612 diff --git a/generator/outputs/models/p5_vae_final_patchgan.pt b/generator/outputs/models/p5_vae_final_patchgan.pt new file mode 100644 index 0000000..975e708 --- /dev/null +++ b/generator/outputs/models/p5_vae_final_patchgan.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:13e3c996bc73e8cf42d7fd9f9a7381c603eafa1474b4f1cc44521bdb269d6dc7 +size 11067346 diff --git a/generator/outputs/models/p5_vae_final_vae.pt b/generator/outputs/models/p5_vae_final_vae.pt new file mode 100644 index 0000000..b8cde2b --- /dev/null +++ b/generator/outputs/models/p5_vae_final_vae.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5cb86d220ab4fbc65f08d71cbb964395f382cde202323e0c117fe5de04c066e4 +size 42444252 diff --git a/generator/outputs/pipeline/20260503T203119.199517+0000.json b/generator/outputs/pipeline/20260503T203119.199517+0000.json new file mode 100644 index 0000000..26723e1 --- /dev/null +++ b/generator/outputs/pipeline/20260503T203119.199517+0000.json @@ -0,0 +1,12 @@ +{ + "created_at": "2026-05-03T20:31:19.199517+00:00", + "config_paths": [ + "generator/configs/phase5/p5_vae.json" + ], + "instance_id": 36081371, + "offer_id": 20017831, + "ssh_host": "ssh1.vast.ai", + "ssh_port": 11370, + "status": "completed", + "remote_workspace": "/workspace/DRL_PROJ" +} \ No newline at end of file diff --git a/generator/outputs/samples/p5_vae/epoch_0010.png b/generator/outputs/samples/p5_vae/epoch_0010.png new file mode 100644 index 0000000..5d74334 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0010.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0010_recon.png b/generator/outputs/samples/p5_vae/epoch_0010_recon.png new file mode 100644 index 0000000..1be75f6 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0010_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0020.png b/generator/outputs/samples/p5_vae/epoch_0020.png new file mode 100644 index 0000000..f56da0e Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0020.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0020_recon.png b/generator/outputs/samples/p5_vae/epoch_0020_recon.png new file mode 100644 index 0000000..7032128 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0020_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0030.png b/generator/outputs/samples/p5_vae/epoch_0030.png new file mode 100644 index 0000000..bffd95b Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0030.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0030_recon.png b/generator/outputs/samples/p5_vae/epoch_0030_recon.png new file mode 100644 index 0000000..eea667e Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0030_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0040.png b/generator/outputs/samples/p5_vae/epoch_0040.png new file mode 100644 index 0000000..0a5003d Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0040.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0040_recon.png b/generator/outputs/samples/p5_vae/epoch_0040_recon.png new file mode 100644 index 0000000..519fa34 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0040_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0050.png b/generator/outputs/samples/p5_vae/epoch_0050.png new file mode 100644 index 0000000..e122f6a Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0050.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0050_recon.png b/generator/outputs/samples/p5_vae/epoch_0050_recon.png new file mode 100644 index 0000000..bc89e43 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0050_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0060.png b/generator/outputs/samples/p5_vae/epoch_0060.png new file mode 100644 index 0000000..6128923 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0060.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0060_recon.png b/generator/outputs/samples/p5_vae/epoch_0060_recon.png new file mode 100644 index 0000000..02f15ad Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0060_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0070.png b/generator/outputs/samples/p5_vae/epoch_0070.png new file mode 100644 index 0000000..e9d5695 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0070.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0070_recon.png b/generator/outputs/samples/p5_vae/epoch_0070_recon.png new file mode 100644 index 0000000..4f12825 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0070_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0080.png b/generator/outputs/samples/p5_vae/epoch_0080.png new file mode 100644 index 0000000..8d52c11 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0080.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0080_recon.png b/generator/outputs/samples/p5_vae/epoch_0080_recon.png new file mode 100644 index 0000000..8a9de16 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0080_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0090.png b/generator/outputs/samples/p5_vae/epoch_0090.png new file mode 100644 index 0000000..5a674a8 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0090.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0090_recon.png b/generator/outputs/samples/p5_vae/epoch_0090_recon.png new file mode 100644 index 0000000..dd3c4a6 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0090_recon.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0100.png b/generator/outputs/samples/p5_vae/epoch_0100.png new file mode 100644 index 0000000..0cfe8c9 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0100.png differ diff --git a/generator/outputs/samples/p5_vae/epoch_0100_recon.png b/generator/outputs/samples/p5_vae/epoch_0100_recon.png new file mode 100644 index 0000000..f30b997 Binary files /dev/null and b/generator/outputs/samples/p5_vae/epoch_0100_recon.png differ