How to save & load models

Nico-Curti / NumPyNet

Neural Networks library in pure numpy

https://nico-curti.github.io/NumPyNet

Other

63 stars 6 forks source link

How to save & load models #8

Open mahearn22 opened 7 months ago

mahearn22 commented 7 months ago

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

mahearn22 commented 7 months ago

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?