Commit 93fa64b0 authored by lucas_miranda's avatar lucas_miranda
Browse files

Modified LSTMs to work with cuDNN implementation

parent f6d81b94
...@@ -384,9 +384,6 @@ if runs > 1: ...@@ -384,9 +384,6 @@ if runs > 1:
for run in range(runs): for run in range(runs):
# To avoid stability issues
tf.keras.backend.clear_session()
run_ID = "{}{}{}{}{}{}_{}".format( run_ID = "{}{}{}{}{}{}_{}".format(
("GMVAE" if variational else "AE"), ("GMVAE" if variational else "AE"),
("P" if predictor > 0 and variational else ""), ("P" if predictor > 0 and variational else ""),
...@@ -505,6 +502,9 @@ for run in range(runs): ...@@ -505,6 +502,9 @@ for run in range(runs):
grouper.predict(input_dict_train[input_type]), axis=1 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 specified (-s > 1), saves the resulting groupings to a dataframe on disk
if runs > 1: if runs > 1:
clust_assignments = pd.DataFrame(clust_assignments) clust_assignments = pd.DataFrame(clust_assignments)
......
Supports Markdown
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