mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Bug fix in save-bad-model code
This commit is contained in:
parent
788c4d97f1
commit
829e4bd4db
@ -861,7 +861,7 @@ def train_one_epoch(
|
||||
params=params,
|
||||
optimizer=optimizer,
|
||||
scheduler=scheduler,
|
||||
sampler=sampler,
|
||||
sampler=train_dl.sampler,
|
||||
scaler=scaler,
|
||||
rank=0)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user