rnnt infer

This commit is contained in:
aky15 2023-04-10 15:54:12 +08:00
parent 4dc3a1b011
commit e01f742a85

View File

@ -301,9 +301,6 @@ class Speech2Text:
feats = speech.unsqueeze(0).to(getattr(torch, self.dtype))
feats_lengths = feats.new_full([1], dtype=torch.long, fill_value=feats.size(1))
if self.asr_model.normalize is not None:
feats, feats_lengths = self.asr_model.normalize(feats, feats_lengths)
feats = to_device(feats, device=self.device)
feats_lengths = to_device(feats_lengths, device=self.device)