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,
|
params=params,
|
||||||
optimizer=optimizer,
|
optimizer=optimizer,
|
||||||
scheduler=scheduler,
|
scheduler=scheduler,
|
||||||
sampler=sampler,
|
sampler=train_dl.sampler,
|
||||||
scaler=scaler,
|
scaler=scaler,
|
||||||
rank=0)
|
rank=0)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user