MichalBusta/E2E-MLT

Error Reloading the model from a checkpoint model

Closed this issue · 1 comments

Hi,

I ran the model for a couple of steps and stopped and tryin the continue training from the old checkpoint model (.h5 file).
The only changes I make is the batch_size parameter from 8 to16 and ocr_batch_size from 256 to 1024. I get the following error -

image

Any solution to this.

Thanks !

No issues I got it working correctly. This was caused because I was meddling with height and width of input size.