Commit d10f27e4 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 9d816a3c
......@@ -134,7 +134,7 @@ else:
"units_lstm": 256,
"units_dense2": 64,
"dropout_rate": 0.25,
"encoding": 32,
"encoding": 8,
"learning_rate": 1e-3,
}
......
......@@ -159,7 +159,7 @@ class SEQ_2_SEQ_GMVAE:
units_lstm=256,
units_dense2=64,
dropout_rate=0.25,
encoding=2,
encoding=16,
learning_rate=1e-3,
loss="ELBO+MMD",
kl_warmup_epochs=0,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment