Testing VAE until it works - v1

2026-05-02 13:11:56 +01:00
parent c7804d2984
commit ec8d4ae336
84 changed files with 9 additions and 1744 deletions
@@ -7,7 +7,6 @@
  "model": "vae",
  "latent_dim": 256,
  "ngf": 64,
-  "free_bits": 0.1,
  "grad_clip": 1.0,
  "sample_interval": 10,
  "fid_interval": 25,
@@ -2,7 +2,7 @@
  "extends": "_base_phase3.json",
  "run_name": "p3_1_vae",
  "lr": 5e-4,
-  "beta_kl": 0.5,
+  "beta_kl": 0.25,
  "lambda_perceptual": 0.0,
  "lambda_adversarial": 0.0
 }
@@ -2,7 +2,7 @@
  "extends": "_base_phase3.json",
  "run_name": "p3_2_vae_perceptual",
  "lr": 5e-4,
-  "beta_kl": 0.1,
+  "beta_kl": 0.25,
  "lambda_perceptual": 0.1,
  "lambda_adversarial": 0.0
 }
@@ -3,7 +3,7 @@
  "run_name": "p3_3_vae_patchgan",
  "lr": 5e-4,
  "lr_d": 1e-4,
-  "beta_kl": 0.05,
+  "beta_kl": 0.25,
  "lambda_perceptual": 0.1,
  "lambda_adversarial": 0.1,
  "ndf_patch": 64