nr: refactor

add new save method

nr: refactor
add new save method
2f4ac40d · Mohammadreza · 915fb560 · 2f4ac40d
Commit 2f4ac40d authored Jul 18, 2021 by Mohammadreza
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

run_auto_encoder.py run_auto_encoder.py +1 -0

No files found.
--- a/run_auto_encoder.py
+++ b/run_auto_encoder.py
@@ -256,6 +256,7 @@ if desired_model == 'conv':
        plt.show()

    # model.save('conv_model.h5')
+    # model.save('conv_model_new.h5', include_optimizer=False)

 elif desired_model == 'dense':