Notas sobre o uso do CTCloss que vem com o pytorch

Quando se trata de projetos de fala ou treinamento LSTM do CRNN, o CTCLoss geralmente é usado. Ao usar uma versão inferior do pytorch, usa-se o Baidu CTCloss. Como o cartão 3090 foi alterado, o pytorch também foi forçado a atualizar para a versão 1.8. Não há como para ser compatível com Baidu CTC, volte para o integrado em torch.nn import CTCLoss,

Verificou-se que o treinamento não previu o conteúdo por muito tempo. Acontece que a diferença do Baidu CTC é que a entrada do CTCloss que vem com o pytorch precisa passar pelo log_softmax. Durante o treinamento, a camada de saída da rede é adicionada com

net(image).log_softmax()

problema resolvido.

 

Acho que você gosta

Origin blog.csdn.net/qq_36276587/article/details/117416054
Recomendado
Clasificación