ingale
ingale
> > pytorch自带的ctcloss在gpu上运行label长度要 >  > > 我自己在训练时也会出现这种错,把超过256的label剔除出去,loss就可以正常计算了。这个问题一直到1.5版本依然存在。 > > 这个256长度的label 是一个batch超过256 还是一张图片中的标签超过256 同问!!!!!!!!!!!!!!! pytorch1.7.1,cuda10.1,还是会有naN出现
训练时候用的什么模型,你就加载什么模型
Error(s) in loading state_dict for CRNN: 加载模型参数的时候就报错了,训练的时候用的什么网络,加载的时候要加载相同的网络,这样参数才对的上
至于GIT主说的用pytorch1.2版本训练没有出现NAN,我猜应该是GIT主的训练集里没有字数特别多的图片和标签,比如说字数大于40个,应该pytorch各个版本都有这样的问题
> 你是pytorh版本1.7.1下安装了第三方CTCLOSS的吗? 我在1.6版面下安装不成功第三方CTCLOSS, 提示ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead. > 运行python setup install, 报的错误 百度一下就找到答案了 https://blog.csdn.net/ShuqiaoS/article/details/88420326 网上有安装第三方CTCLOSS的博客,就是安装过程是坑较多,多看看issues
> 你是pytorh版本1.7.1下安装了第三方CTCLOSS的吗? 我在1.6版面下安装不成功第三方CTCLOSS, 提示ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead. > 运行python setup install, 报的错误 https://blog.csdn.net/xiao13mm/article/details/112426460
我改成了.cuda(),不知道效果怎么样
> @dc-chengchao 您好,您说的280指的是生成的图像的宽度吧?但是训练的时候又resize成宽度为160了,为什么要这样做呢,这样不是把原先比较正常的形状的文字压瘦了?我看到github上好几个作者都是这样做的,不明白出于什么原因。 我直接改代码了,图片原比例280进入模型训练,正常训练
> > CRNN的训练代码,您用的是哪个训练代码啊?能发下github网址吗? > > 你好,请问你找到训练代码了吗? https://github.com/Sierkinhane/CRNN_Chinese_Characters_Rec