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
8436c2c44b
commit
0d21e11062
Binary file not shown.
@ -967,7 +967,7 @@ def train_one_epoch(
|
||||
optimizer: torch.optim.Optimizer or [torch.optim.Optimizer, torch.optim.Optimizer],
|
||||
scheduler: LRSchedulerType or [LRSchedulerType, LRSchedulerType],
|
||||
sp: spm.SentencePieceProcessor,
|
||||
train_dl: torch.utils.data.DataLoader,
|
||||
train_dl: torch.utils.data.DataLoader or [torch.utils.data.DataLoader, torch.utils.data.DataLoader],
|
||||
valid_dl: torch.utils.data.DataLoader,
|
||||
scaler: GradScaler,
|
||||
model_avg: Optional[nn.Module] = None,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user