Commit 0210c812 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented KL and MMD warmup on SEQ2SEQ_VAEP in models.py

parent 942674c7
......@@ -383,6 +383,10 @@ class SEQ_2_SEQ_VAEP:
self.kl_warmup = kl_warmup_epochs
self.mmd_warmup = mmd_warmup_epochs
assert (
"ELBO" in self.loss or "MMD" in self.loss
), "loss must be one of ELBO, MMD or ELBO+MMD (default)"
def build(self):
# Encoder Layers
Model_E0 = tf.keras.layers.Conv1D(
......
Markdown is supported
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