Compute loss value from buffer sequences. Args: input (torch.Tensor): Input ids. (batch, len) hidden (torch.Tensor): Target ids. (batch, len)
(self, text: torch.Tensor, text_lengths: torch.Tensor, **kwargs)
| 111 | self.jieba_usr_dict = jieba |
| 112 | |
| 113 | def punc_forward(self, text: torch.Tensor, text_lengths: torch.Tensor, **kwargs): |
| 114 | """Compute loss value from buffer sequences. |
| 115 | |
| 116 | Args: |
| 117 | input (torch.Tensor): Input ids. (batch, len) |
| 118 | hidden (torch.Tensor): Target ids. (batch, len) |
| 119 | |
| 120 | """ |
| 121 | x = self.embed(text) |
| 122 | # mask = self._target_mask(input) |
| 123 | h, _, _ = self.encoder(x, text_lengths) |
| 124 | y = self.decoder(h) |
| 125 | return y, None |
| 126 | |
| 127 | def with_vad(self): |
| 128 | """With vad.""" |