Bug fix in save-bad-model code

This commit is contained in:
Daniel Povey 2022-12-21 15:17:26 +08:00
parent 788c4d97f1
commit 829e4bd4db

View File

@ -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)