Increased default dimensionality of latent space

69196db8 · lucas_miranda · 2f603a5e · 69196db8
Commit 69196db8 authored Jun 04, 2021 by lucas_miranda
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

deepof/models.py deepof/models.py +2 -2

No files found.
--- a/deepof/models.py
+++ b/deepof/models.py
@@ -136,8 +136,8 @@ class GMVAE:
            "clipvalue": 1.0,
            "dense_activation": "relu",
            "dense_layers_per_branch": 3,
-            "dropout_rate": 0.05,
-            "learning_rate": 1e-3,
+            "dropout_rate": 0.1,
+            "learning_rate": 1e-4,
            "units_conv": 64,
            "units_dense2": 32,
            "units_gru": 128,