CHINESE-OCR
CHINESE-OCR copied to clipboard
pytorch 训练问题用pytorch自带的CTCLoss
我想问下warpctc_pytorch和pytorch两者的CTCLoss传参有有什么区别,代码中要用pytorch自带的CTCLoss应该如何调整
参考这篇博客和作者实现的crnn代码,https://blog.csdn.net/Sierkinhane/article/details/82857572,用pytorch自带的话直接注释掉from warpctc_pytorch import CTCLoss
,然后把141行改为criterion = torch.nn.CTCLoss(reduction='sum')
就行了