From 93fa64b05933ea969bf2489e7c73b683b5383a96 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Mon, 13 Jul 2020 09:04:48 +0200
Subject: [PATCH] Modified LSTMs to work with cuDNN implementation

---
 model_training.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/model_training.py b/model_training.py
index 1eb8b180..39e89246 100644
--- a/model_training.py
+++ b/model_training.py
@@ -384,9 +384,6 @@ if runs > 1:
 
 for run in range(runs):
 
-    # To avoid stability issues
-    tf.keras.backend.clear_session()
-
     run_ID = "{}{}{}{}{}{}_{}".format(
         ("GMVAE" if variational else "AE"),
         ("P" if predictor > 0 and variational else ""),
@@ -505,6 +502,9 @@ for run in range(runs):
                 grouper.predict(input_dict_train[input_type]), axis=1
             )
 
+    # To avoid stability issues
+    tf.keras.backend.clear_session()
+
 # If specified (-s > 1), saves the resulting groupings to a dataframe on disk
 if runs > 1:
     clust_assignments = pd.DataFrame(clust_assignments)
-- 
GitLab