jalf/DRL_PROJ
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Trying a few different VAE settings v2

Browse Source
This commit is contained in:
Johnny Fernandes
2026-05-02 02:20:40 +01:00
parent 5c047e4de7
commit ec680d0288
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/configs/phase3/p3_1_vae.json
+1 -1
View File
@@ -2,7 +2,7 @@
"extends": "_base_phase3.json", "extends": "_base_phase3.json",
"run_name": "p3_1_vae", "run_name": "p3_1_vae",
"lr": 5e-4, "lr": 5e-4,
"beta_kl": 0.005, "beta_kl": 0.5,
"lambda_perceptual": 0.0, "lambda_perceptual": 0.0,
"lambda_adversarial": 0.0 "lambda_adversarial": 0.0
} }
generator/configs/phase3/p3_2_vae_perceptual.json
+1 -1
View File
@@ -2,7 +2,7 @@
"extends": "_base_phase3.json", "extends": "_base_phase3.json",
"run_name": "p3_2_vae_perceptual", "run_name": "p3_2_vae_perceptual",
"lr": 5e-4, "lr": 5e-4,
"beta_kl": 0.005, "beta_kl": 0.1,
"lambda_perceptual": 0.1, "lambda_perceptual": 0.1,
"lambda_adversarial": 0.0 "lambda_adversarial": 0.0
} }
generator/configs/phase3/p3_3_vae_patchgan.json
+1 -1
View File
@@ -3,7 +3,7 @@
"run_name": "p3_3_vae_patchgan", "run_name": "p3_3_vae_patchgan",
"lr": 5e-4, "lr": 5e-4,
"lr_d": 1e-4, "lr_d": 1e-4,
"beta_kl": 0.005, "beta_kl": 0.05,
"lambda_perceptual": 0.1, "lambda_perceptual": 0.1,
"lambda_adversarial": 0.1, "lambda_adversarial": 0.1,
"ndf_patch": 64 "ndf_patch": 64