search

bes-dev / crnn-pytorch

Pytorch implementation of OCR system using CRNN + CTCLoss
BSD 2-Clause "Simplified" License
216 stars 55 forks source link

Evaluation on CPU #12

Closed cbasavaraj closed 4 years ago

cbasavaraj commented 6 years ago

Hi, I am getting poor performance when I test a trained model on CPU (but the checkpoint's working well on GPU as per validation numbers). Details here: https://discuss.pytorch.org/t/different-results-on-cpu-and-gpu/22289/3 Appreciate any feedback, thanks.

bes-dev commented 6 years ago

Hi, Did you check that all layers has been loaded from checkpoint? It's look like that some keys from snapshot not match with keys from network and some layers initialize with random weights on CPU.