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,
optimizer=optimizer,
scheduler=scheduler,
sampler=sampler,
sampler=train_dl.sampler,
scaler=scaler,
rank=0)