From 829e4bd4dbd464634b9194c0d98d1d3fbdfd230e Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 21 Dec 2022 15:17:26 +0800 Subject: [PATCH] Bug fix in save-bad-model code --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index 0e9fe1423..72cec1328 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -861,7 +861,7 @@ def train_one_epoch( params=params, optimizer=optimizer, scheduler=scheduler, - sampler=sampler, + sampler=train_dl.sampler, scaler=scaler, rank=0)