CHINESE-OCR icon indicating copy to clipboard operation
CHINESE-OCR copied to clipboard

pytorch 训练问题用pytorch自带的CTCLoss

Open psw13583112 opened this issue 5 years ago • 1 comments

我想问下warpctc_pytorch和pytorch两者的CTCLoss传参有有什么区别,代码中要用pytorch自带的CTCLoss应该如何调整

psw13583112 avatar Jun 18 '19 02:06 psw13583112

参考这篇博客和作者实现的crnn代码,https://blog.csdn.net/Sierkinhane/article/details/82857572,用pytorch自带的话直接注释掉from warpctc_pytorch import CTCLoss,然后把141行改为criterion = torch.nn.CTCLoss(reduction='sum')就行了

Develkone avatar Aug 13 '19 09:08 Develkone