VAE fix w/ new results

2026-05-02 00:32:45 +01:00
parent 1a7f67ab9c
commit bac52bc15e
90 changed files with 1197 additions and 1106 deletions
@@ -7,6 +7,7 @@
  "model": "vae",
  "latent_dim": 256,
  "ngf": 64,
+  "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.25,
  "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.25,
  "lambda_perceptual": 0.1,
  "lambda_adversarial": 0.0
 }
@@ -1,10 +1,10 @@
 {
  "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.25,
  "lambda_perceptual": 0.1,
-  "lambda_adversarial": 0.1,
+  "lambda_adversarial": 0.01,
  "ndf_patch": 64
 }
@@ -6,5 +6,6 @@
  "subsample": 1.0,
  "sample_interval": 10,
  "fid_interval": 25,
-  "fid_n_real": 5000
+  "fid_n_real": 5000,
+  "num_workers": 2
 }