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