RuntimeError: CUDA error: an illegal memory access was encountered

CRNN训练出现了RuntimeError: CUDA error: an illegal memory access was encountered错误。

使用的PyTorch 1.1,自带CTCLoss函数。

查阅资料,发现既不是字典少字,也不是空格以及换行符的原因。

发现最后的错误来源是因为CTC函数,于是更换成了https://github.com/SeanNaren/warp-ctc的函数,一切正常了。。。

 

【2019年10月23日】PyTorch更新了1.3版本,解决了torch.nn.CTCLoss在训练过程中出现Nan的问题?应该还是需要修改CRNN的训练代码。。。

 

【参考资料】https://blog.csdn.net/baidu_36639782/article/details/89405743

你可能感兴趣的:(深度学习,OCR)