Add connectionist temporal classification (CTC) loss algorithm - #11240
ZJUGuoShuai wants to merge 2 commits into
Conversation
There was a problem hiding this comment.
A couple suggestions to clarify the documentation:
-
PyTorch cites Graves et al for its implementation. Since your implementation is also based on this paper, please add it as a reference as well.
-
Please add a short paragraph explaining how the loss is actually calculated. This should explain some general questions about your variables (What is
blank? What isalpha, and why is it calculated using DP?) so that the reader understands what is being calculated. I ask for this because this repository is meant for educational purposes, so we want readers to understand how and why the implementation works.Also, in your implementation you use
np.logaddexpfor log-probabilities when calculatingalpharather than calculating probabilities directly. Since this differs from the definitions in the Graves et al paper, please be sure to note this implementation detail in your explanation.

Describe your change:
Checklist: