Commit 3a9eca76 authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed epochs default for model training

parent 67949328
Pipeline #99994 passed with stages
in 22 minutes and 49 seconds
...@@ -426,8 +426,17 @@ def autoencoder_fitting( ...@@ -426,8 +426,17 @@ def autoencoder_fitting(
], ],
) )
if not os.path.exists(os.path.join(output_path, "trained_weights")):
os.makedirs(os.path.join(output_path, "trained_weights"))
if save_weights: if save_weights:
ae.save_weights("{}_final_weights.h5".format(run_ID)) ae.save_weights(
os.path.join(
"{}".format(output_path),
"trained_weights",
"{}_final_weights.h5".format(run_ID),
)
)
else: else:
...@@ -473,7 +482,7 @@ def autoencoder_fitting( ...@@ -473,7 +482,7 @@ def autoencoder_fitting(
) )
if not os.path.exists(os.path.join(output_path, "trained_weights")): if not os.path.exists(os.path.join(output_path, "trained_weights")):
os.makedirs("trained_weights") os.makedirs(os.path.join(output_path, "trained_weights"))
if save_weights: if save_weights:
ae.save_weights( ae.save_weights(
......
Markdown is supported
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