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(
],
)
if not os.path.exists(os.path.join(output_path, "trained_weights")):
os.makedirs(os.path.join(output_path, "trained_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:
......@@ -473,7 +482,7 @@ def autoencoder_fitting(
)
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:
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