From ec680d02884cab143218f34dfff6f7ebebe526a4 Mon Sep 17 00:00:00 2001 From: Johnny Fernandes Date: Sat, 2 May 2026 02:20:40 +0100 Subject: [PATCH] Trying a few different VAE settings v2 --- generator/configs/phase3/p3_1_vae.json | 2 +- generator/configs/phase3/p3_2_vae_perceptual.json | 2 +- generator/configs/phase3/p3_3_vae_patchgan.json | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/generator/configs/phase3/p3_1_vae.json b/generator/configs/phase3/p3_1_vae.json index 0951564..61e459b 100644 --- a/generator/configs/phase3/p3_1_vae.json +++ b/generator/configs/phase3/p3_1_vae.json @@ -2,7 +2,7 @@ "extends": "_base_phase3.json", "run_name": "p3_1_vae", "lr": 5e-4, - "beta_kl": 0.005, + "beta_kl": 0.5, "lambda_perceptual": 0.0, "lambda_adversarial": 0.0 } diff --git a/generator/configs/phase3/p3_2_vae_perceptual.json b/generator/configs/phase3/p3_2_vae_perceptual.json index 93173e8..49f7d8a 100644 --- a/generator/configs/phase3/p3_2_vae_perceptual.json +++ b/generator/configs/phase3/p3_2_vae_perceptual.json @@ -2,7 +2,7 @@ "extends": "_base_phase3.json", "run_name": "p3_2_vae_perceptual", "lr": 5e-4, - "beta_kl": 0.005, + "beta_kl": 0.1, "lambda_perceptual": 0.1, "lambda_adversarial": 0.0 } diff --git a/generator/configs/phase3/p3_3_vae_patchgan.json b/generator/configs/phase3/p3_3_vae_patchgan.json index 8fa823c..7be6288 100644 --- a/generator/configs/phase3/p3_3_vae_patchgan.json +++ b/generator/configs/phase3/p3_3_vae_patchgan.json @@ -3,7 +3,7 @@ "run_name": "p3_3_vae_patchgan", "lr": 5e-4, "lr_d": 1e-4, - "beta_kl": 0.005, + "beta_kl": 0.05, "lambda_perceptual": 0.1, "lambda_adversarial": 0.1, "ndf_patch": 64