Help With Saving and Loading Model

Hi, I am doing optical character recognition on my own dataset, consisting of around 17k images of 11 classes (0-9 as well as $). I can train the model no problem, only 2 epochs for now as loss goes down very quickly and it works perfectly immediately after training. The issue is that I try and save the model, then try and load the model, and it is like I never loaded it at all. The classifications are terrible and it barely gets 1 or 2 of the 16 images used for inference testing (completely random).

I’m sure I am doing something wrong, but I just can’t figure out what.

I’ve been following [this]( guide from keras, and [this]( is my GitHub repo with my specific code.

submitted by /u/BoopJoop01
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *