search

MichalBusta / E2E-MLT

E2E-MLT - an Unconstrained End-to-End Method for Multi-Language Scene Text
MIT License
291 stars 84 forks source link

Error Reloading the model from a checkpoint model #74

Closed mohammedayub44 closed 3 years ago

mohammedayub44 commented 3 years ago

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 !

mohammedayub44 commented 3 years ago

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