Reduce target beta

hellomoto-ai · May 10, 2019 · 4cc2090 · 4cc2090
1 parent 611af39
commit 4cc2090
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sp_vae_gan/trainer.py b/sp_vae_gan/trainer.py
@@ -52,7 +52,7 @@ def __init__(
             device, output_dir,
             initial_beta=10.0,
             beta_step=0.1,
-            target_kld=0.2,
+            target_kld=0.1,
     ):
         self.model = model.float().to(device)
         self.train_loader = train_loader