How to save & load models

Open mahearn22 opened this issue 1 year ago • 1 comments

Hello,

Forgive me as I am still learning, but I am looking for some assistance please, once I have trained a CNN model how do I then save/load the model and make predictions on images? I try to use the model.save_model and model.save_weights, but how should the network initialisation be? Any examples of doing this once a model has been trained would be much appreciated.

thanks

Feb 19 '24 23:02 mahearn22

Also when trying to load_weights I get an error on load_weights for batchnorm_layer: when trying to reshape the bias into a (256,27,27,32) shape. Any ideas on why and how to fix?

Feb 20 '24 22:02 mahearn22