minor fixes
This commit is contained in:
parent
b34eafa500
commit
bb37894fa9
@ -730,7 +730,7 @@ def train_one_epoch(
|
|||||||
scheduler: LRSchedulerType,
|
scheduler: LRSchedulerType,
|
||||||
graph_compiler: CharCtcTrainingGraphCompiler,
|
graph_compiler: CharCtcTrainingGraphCompiler,
|
||||||
train_dl: torch.utils.data.DataLoader,
|
train_dl: torch.utils.data.DataLoader,
|
||||||
datatang_train_dl: torch.utils.data.DataLoader,
|
datatang_train_dl: Optional[torch.utils.data.DataLoader],
|
||||||
valid_dl: torch.utils.data.DataLoader,
|
valid_dl: torch.utils.data.DataLoader,
|
||||||
rng: random.Random,
|
rng: random.Random,
|
||||||
scaler: GradScaler,
|
scaler: GradScaler,
|
||||||
|
|||||||
Reference in New Issue
Block a user