from local

This commit is contained in:
dohe0342 2023-02-25 15:51:30 +09:00
parent c599a30b07
commit 3d87d438fc
3 changed files with 1 additions and 0 deletions

View File

@ -263,6 +263,7 @@ class Transformer(nn.Module):
sos_id: int, sos_id: int,
eos_id: int, eos_id: int,
warmup: float = 1.0, warmup: float = 1.0,
return_output: bool = False,
) -> torch.Tensor: ) -> torch.Tensor:
""" """
Args: Args: