90092a23db57abba13808dbf96100e07fc4797ac,kraken/lib/ctc.py,_CTC,backward,#_CTC#,106

Before Change

        return loss

    def backward(self, grad_output):
        return self.grads, None, None, None


class CTCCriterion(Module):
    r

After Change

        label_prob = self.label_probability(self.yseq.shape[2],
                                            self.path,
                                            self.path_length,
                                            np.exp(self.prob_trans - total_probability[:, None]))
        self.yseq -= label_prob
        // mask
        self.yseq *= (np.arange(len(self.yseq))[:, None] < self.input_length)[..., None]
        return torch.tensor(self.yseq).permute(1, 2, 0), None


class CTCCriterion(Module):
    r

Instances