From ae3631d2453cd786c006e9d6ffda26f616c8980c Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Tue, 13 Apr 2021 17:16:14 +0200
Subject: [PATCH] Added extra branch to main autoencoder for rule_based
 prediction

---
 deepof/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepof/models.py b/deepof/models.py
index 7bf5eb54..86f8888f 100644
--- a/deepof/models.py
+++ b/deepof/models.py
@@ -800,3 +800,4 @@ class SEQ_2_SEQ_GMVAE:
 #       - Check usefulness of stateful sequential layers! (stateful=True in the LSTMs)
 #       - Investigate full covariance matrix approximation for the latent space! (details on tfp course) :)
 #       - Explore expanding the event dims of the final reconstruction layer
+#       - Think about gradient penalty to avoid mode collapse (as in WGAN-GP)
-- 
GitLab