Preview of phase 2-5 implementation; needs a full check

2026-04-30 13:10:33 +01:00
parent 6e32001ebc
commit 7417267117
35 changed files with 3605 additions and 115 deletions
@@ -0,0 +1,13 @@
+{
+  "epochs": 100,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": "hflip",
+  "image_size": 64,
+  "model": "vae",
+  "latent_dim": 256,
+  "ngf": 64,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_1_vae",
+  "lr": 1e-3,
+  "beta_kl": 1.0,
+  "lambda_perceptual": 0.0,
+  "lambda_adversarial": 0.0
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_2_vae_perceptual",
+  "lr": 1e-3,
+  "beta_kl": 0.0001,
+  "lambda_perceptual": 0.1,
+  "lambda_adversarial": 0.0
+}
@@ -0,0 +1,10 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_3_vae_patchgan",
+  "lr": 1e-3,
+  "lr_d": 1e-4,
+  "beta_kl": 0.0001,
+  "lambda_perceptual": 0.1,
+  "lambda_adversarial": 0.1,
+  "ndf_patch": 64
+}