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