from local

This commit is contained in:
dohe0342 2023-02-14 17:49:01 +09:00
parent f975150f19
commit d90ec45e33
3 changed files with 1 additions and 1 deletions

View File

@ -236,7 +236,7 @@ class Transformer(nn.Module):
return x, mask
def ctc_output(self, x: torch.Tensor) -> torch.Tensor:
def ctc_output(self, x: torch.Tensor, log_prob: bool = True) -> torch.Tensor:
"""
Args:
x: