diff --git a/deepof/train_utils.py b/deepof/train_utils.py index 36f2fbab311892622529652951f9b29c6251e6ed..2a161cfcc9b20aba414054a6e149169ac36c62c5 100644 --- a/deepof/train_utils.py +++ b/deepof/train_utils.py @@ -432,7 +432,9 @@ def autoencoder_fitting( if save_weights: ae.save_weights( os.path.join( - "{}".format(output_path), "{}_final_weights.h5".format(run_ID) + "{}".format(output_path), + "trained_weights", + "{}_final_weights.h5".format(run_ID), ) )