From 04fc4e4f4a6637c2ef113b6a8afbd9533830d2ed Mon Sep 17 00:00:00 2001 From: pkufool Date: Mon, 11 Apr 2022 15:41:45 +0800 Subject: [PATCH] Minor fixes --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index 20a111dd5..04a9aebb4 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -856,7 +856,7 @@ def run(rank, world_size, args): valid_cuts += librispeech.dev_other_cuts() valid_dl = librispeech.valid_dataloaders(valid_cuts) - if not params.print_diagnostics and not params.use_fp16: + if not params.print_diagnostics: scan_pessimistic_batches_for_oom( model=model, train_dl=train_dl,