mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
d0f8e6067d
commit
685d39918f
Binary file not shown.
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user