From a92133ef960a43d3e9f4834594acad2051c9aa22 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 31b85d53c..577231995 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -869,7 +869,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,