from local

This commit is contained in:
dohe0342 2022-12-10 13:57:06 +09:00
parent d0f8e6067d
commit 685d39918f
2 changed files with 1 additions and 1 deletions

View File

@ -878,7 +878,7 @@ def train_one_epoch(
params: AttributeDict,
model: Union[nn.Module, DDP],
optimizer: torch.optim.Optimizer or [torch.optim.Optimizer, torch.optim.Optimizer],
scheduler: LRSchedulerType,
scheduler: LRSchedulerType or [LRSchedulerType, LRSchedulerType],
sp: spm.SentencePieceProcessor,
train_dl: torch.utils.data.DataLoader,
valid_dl: torch.utils.data.DataLoader,