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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值