VAE refactor

2026-05-02 00:32:45 +01:00
parent 1a7f67ab9c
commit c7804d2984
43 changed files with 1140 additions and 1064 deletions
@@ -7,6 +7,8 @@
  "model": "vae",
  "latent_dim": 256,
  "ngf": 64,
+  "free_bits": 0.1,
+  "grad_clip": 1.0,
  "sample_interval": 10,
  "fid_interval": 25,
  "fid_n_real": 5000
@@ -1,8 +1,8 @@
 {
  "extends": "_base_phase3.json",
  "run_name": "p3_1_vae",
-  "lr": 1e-3,
-  "beta_kl": 1.0,
+  "lr": 5e-4,
+  "beta_kl": 0.5,
  "lambda_perceptual": 0.0,
  "lambda_adversarial": 0.0
 }
@@ -1,8 +1,8 @@
 {
  "extends": "_base_phase3.json",
  "run_name": "p3_2_vae_perceptual",
-  "lr": 1e-3,
-  "beta_kl": 0.0001,
+  "lr": 5e-4,
+  "beta_kl": 0.1,
  "lambda_perceptual": 0.1,
  "lambda_adversarial": 0.0
 }
@@ -1,9 +1,9 @@
 {
  "extends": "_base_phase3.json",
  "run_name": "p3_3_vae_patchgan",
-  "lr": 1e-3,
+  "lr": 5e-4,
  "lr_d": 1e-4,
-  "beta_kl": 0.0001,
+  "beta_kl": 0.05,
  "lambda_perceptual": 0.1,
  "lambda_adversarial": 0.1,
  "ndf_patch": 64