Hi Richard,
I’m implementing a torch nn.Criterion https://github.com/facebookresearch/wav2letter/blob/master/wav2letter/batchautosegcriterionc.lua 19 in pytorch.
It requires updating parameters in the backward part.