diff --git a/generator/outputs/logs/p5_ddpm.json b/generator/outputs/logs/p5_ddpm.json new file mode 100644 index 0000000..a463fb9 --- /dev/null +++ b/generator/outputs/logs/p5_ddpm.json @@ -0,0 +1,232 @@ +{ + "run_name": "p5_ddpm", + "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_ddpm", + "model": "ddpm", + "epochs": 400, + "augment": "hflip", + "image_size": 64, + "T": 1000, + "noise_schedule": "cosine", + "pred_type": "v", + "base_ch": 192, + "ch_mult": [ + 1, + 2, + 2, + 2 + ], + "attn_resolutions": [ + 32, + 16, + 8 + ], + "num_res_blocks": 2, + "dropout": 0.1, + "lr": 0.0002, + "ddim_steps": 100, + "max_train_hours": 24, + "fid_patience": 2 + }, + "history": { + "loss": [ + 0.1361132180557037, + 0.08347086386325267, + 0.07713027890676107, + 0.0735087690827174, + 0.07138374625331062, + 0.07048111248952456, + 0.0699430058001835, + 0.06959152534508552, + 0.0687942563309374, + 0.06749710015570506, + 0.06797479636744301, + 0.06707314404253012, + 0.06645634254583946, + 0.06616808988281295, + 0.06632597825657098, + 0.06543161581533077, + 0.06567037307744862, + 0.06560073971238911, + 0.06487136873870324, + 0.06486488640722302, + 0.06474818710365102, + 0.0648785232542417, + 0.06409876514226198, + 0.06419924900540684, + 0.06415374198148394, + 0.06415291013530432, + 0.06390660826880963, + 0.06351337538888821, + 0.0637402735157018, + 0.06242381525026937, + 0.0636787190157761, + 0.06266482662345864, + 0.0638032130514964, + 0.06235016199449698, + 0.06285461769081079, + 0.06259892232970804, + 0.06228241114280163, + 0.06305973536661293, + 0.06274244885764316, + 0.062076514268405415, + 0.06195374166704396, + 0.06290199403834139, + 0.06198460275195857, + 0.0624536736908122, + 0.06327703114376108, + 0.061569202508236096, + 0.0628280364550077, + 0.06169611253003534, + 0.06185008156407847, + 0.06172168297040412, + 0.061782848997375905, + 0.06151117438561896, + 0.061827609793115884, + 0.061956947136065386, + 0.06214875884704355, + 0.062044317284837745, + 0.061401939210601345, + 0.06086002392933155, + 0.061775257596029684, + 0.061602149421396926, + 0.06117732557030315, + 0.061202412471175194, + 0.061622718205818765, + 0.06128816264402917, + 0.0614978774315399, + 0.061346558766423635, + 0.06128153991368082, + 0.06116244996077994, + 0.06111229184823923, + 0.06107063669488471, + 0.06084184085106493, + 0.060991186720247455, + 0.06110310762260969, + 0.060644112224889614, + 0.06105265081024323, + 0.060728371310501524, + 0.0612431171660622, + 0.06059174603408473, + 0.06193933507793734, + 0.06114670591285595, + 0.06008413203187987, + 0.060815918625483655, + 0.06038160178746678, + 0.060638460115744516, + 0.06106071204392829, + 0.060512709447270274, + 0.05989936954126908, + 0.06053547763353229, + 0.060993485152721405, + 0.06054355477134132, + 0.060113425056139626, + 0.060656892772541084, + 0.06028386112302542, + 0.06064964496554472, + 0.060564143551338434, + 0.06023223410782396, + 0.0606433203141404, + 0.06019309561094667, + 0.060663971755430736, + 0.06061358017544461, + 0.06051926234434558, + 0.060519506510060564, + 0.05998779779188653, + 0.060273363700725585, + 0.060976108679404624, + 0.05960293089509265, + 0.06006794761165849, + 0.06001491510333159, + 0.059924575054429025, + 0.060125686371555694, + 0.06028291987430336, + 0.06011252240556428, + 0.0603115351504495, + 0.059780152243936166, + 0.05937349914899494, + 0.060589286212164625, + 0.05928194311917083, + 0.0603618670700707, + 0.05988672448115216, + 0.06036584552886904, + 0.05977793929413853, + 0.0594497075320309, + 0.059364209859034955, + 0.059953983116926804, + 0.059721899139257066, + 0.06011831242209061, + 0.059677510808866754, + 0.059987764399594225, + 0.060053561774329245, + 0.059788390898551695, + 0.05936959649348616, + 0.059900946533068634, + 0.05933099720053948, + 0.05939908983169967, + 0.0593241080084545, + 0.05961700674528495, + 0.05928131563859618, + 0.05942184386504257, + 0.060225125601212695, + 0.0589471840673787, + 0.06034273053240827, + 0.05934861272923712, + 0.05955804113107614, + 0.05946602335629555, + 0.05946855390301117, + 0.05917746134293385, + 0.05913895009579057, + 0.060033635260202944, + 0.05908472824873578, + 0.060041244786519274, + 0.05998928384043467, + 0.059578305993897795, + 0.05958271934849839, + 0.05902576439369183, + 0.06004863530875017, + 0.058922704555985764, + 0.059662471589051254, + 0.05893384579282541, + 0.05959883256632294, + 0.05884528258003485, + 0.05999988574598335, + 0.059614969194572195, + 0.05979255826459227, + 0.05972598887916304, + 0.05883011880179501, + 0.05925349895165771, + 0.059959421220880285, + 0.05954511561550391, + 0.05976989628890386, + 0.059382665185974196, + 0.059384163946677476, + 0.059122750774408, + 0.05936474042634169, + 0.059725822403262824, + 0.05970647112012674 + ], + "fid": { + "25": 312.4023742675781, + "50": 174.07015991210938, + "75": 81.27722930908203, + "100": 34.90231704711914, + "125": 19.03875160217285, + "150": 19.847490310668945, + "175": 20.106101989746094 + }, + "train_time_s": 40025.90805530548 + }, + "n_params": 64770819 +} \ No newline at end of file diff --git a/generator/outputs/models/p5_ddpm_best_ema.pt b/generator/outputs/models/p5_ddpm_best_ema.pt new file mode 100644 index 0000000..bd860fe --- /dev/null +++ b/generator/outputs/models/p5_ddpm_best_ema.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:82e93ef9b5858cfabe6ce19d51d07bf921e07237a9f0031c9d76e0921e8f0b5b +size 259190319 diff --git a/generator/outputs/models/p5_ddpm_best_unet.pt b/generator/outputs/models/p5_ddpm_best_unet.pt new file mode 100644 index 0000000..4d0ad89 --- /dev/null +++ b/generator/outputs/models/p5_ddpm_best_unet.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:413a9732d3deb92f266ec38fe41b711cc4bea39a1dc4c9ecd0d48bbfedfeff17 +size 259195922 diff --git a/generator/outputs/models/p5_ddpm_final_ema.pt b/generator/outputs/models/p5_ddpm_final_ema.pt new file mode 100644 index 0000000..b245f94 --- /dev/null +++ b/generator/outputs/models/p5_ddpm_final_ema.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a7a6952b1246976daed85e1afd7981e2e268183f7e94d399bd22f48b8b96b2a0 +size 259190610 diff --git a/generator/outputs/models/p5_ddpm_final_unet.pt b/generator/outputs/models/p5_ddpm_final_unet.pt new file mode 100644 index 0000000..f573c2e --- /dev/null +++ b/generator/outputs/models/p5_ddpm_final_unet.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0cdc3afef87a221960a195a6cfd491bb72b162be1fa5d73110da38e16fe56fd9 +size 259196213 diff --git a/generator/outputs/pipeline/20260503T225620.930335+0000.json b/generator/outputs/pipeline/20260503T225620.930335+0000.json new file mode 100644 index 0000000..11fa3f0 --- /dev/null +++ b/generator/outputs/pipeline/20260503T225620.930335+0000.json @@ -0,0 +1,12 @@ +{ + "created_at": "2026-05-03T22:56:20.930335+00:00", + "config_paths": [ + "generator/configs/phase5/p5_ddpm.json" + ], + "instance_id": 36086207, + "offer_id": 31977988, + "ssh_host": "ssh8.vast.ai", + "ssh_port": 16206, + "status": "completed", + "remote_workspace": "/workspace/DRL_PROJ" +} \ No newline at end of file diff --git a/generator/outputs/samples/p5_ddpm/epoch_0010.png b/generator/outputs/samples/p5_ddpm/epoch_0010.png new file mode 100644 index 0000000..415023d Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0010.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0020.png b/generator/outputs/samples/p5_ddpm/epoch_0020.png new file mode 100644 index 0000000..3eea8e9 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0020.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0030.png b/generator/outputs/samples/p5_ddpm/epoch_0030.png new file mode 100644 index 0000000..8d2f262 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0030.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0040.png b/generator/outputs/samples/p5_ddpm/epoch_0040.png new file mode 100644 index 0000000..3bfe7d3 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0040.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0050.png b/generator/outputs/samples/p5_ddpm/epoch_0050.png new file mode 100644 index 0000000..fff0920 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0050.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0060.png b/generator/outputs/samples/p5_ddpm/epoch_0060.png new file mode 100644 index 0000000..6a92228 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0060.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0070.png b/generator/outputs/samples/p5_ddpm/epoch_0070.png new file mode 100644 index 0000000..22dc22e Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0070.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0080.png b/generator/outputs/samples/p5_ddpm/epoch_0080.png new file mode 100644 index 0000000..43b7006 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0080.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0090.png b/generator/outputs/samples/p5_ddpm/epoch_0090.png new file mode 100644 index 0000000..9848661 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0090.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0100.png b/generator/outputs/samples/p5_ddpm/epoch_0100.png new file mode 100644 index 0000000..3ee6c1b Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0100.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0110.png b/generator/outputs/samples/p5_ddpm/epoch_0110.png new file mode 100644 index 0000000..4acd14a Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0110.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0120.png b/generator/outputs/samples/p5_ddpm/epoch_0120.png new file mode 100644 index 0000000..8cfa459 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0120.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0130.png b/generator/outputs/samples/p5_ddpm/epoch_0130.png new file mode 100644 index 0000000..659d2c1 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0130.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0140.png b/generator/outputs/samples/p5_ddpm/epoch_0140.png new file mode 100644 index 0000000..dc52aa1 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0140.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0150.png b/generator/outputs/samples/p5_ddpm/epoch_0150.png new file mode 100644 index 0000000..5a532a3 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0150.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0160.png b/generator/outputs/samples/p5_ddpm/epoch_0160.png new file mode 100644 index 0000000..174332b Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0160.png differ diff --git a/generator/outputs/samples/p5_ddpm/epoch_0170.png b/generator/outputs/samples/p5_ddpm/epoch_0170.png new file mode 100644 index 0000000..ae3a2a4 Binary files /dev/null and b/generator/outputs/samples/p5_ddpm/epoch_0170.png differ