From 5e7e1a350eaa0ec73efabd918bcaedce36b44e5e Mon Sep 17 00:00:00 2001 From: yfy62 Date: Wed, 26 Apr 2023 17:45:00 +0800 Subject: [PATCH] Fit type hint --- egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py b/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py index 32998a8de..8edbf8500 100755 --- a/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py @@ -564,7 +564,7 @@ def save_checkpoint( def compute_loss( params: AttributeDict, - model: nn.Module, + model: Union[nn.Module, DDP], sp: spm.SentencePieceProcessor, pl: UniqLexicon, batch: dict, @@ -948,7 +948,7 @@ def run(rank, world_size, args): valid_cuts = gigaspeech.dev_cuts() valid_dl = gigaspeech.valid_dataloaders(valid_cuts) - if 0 and not params.print_diagnostics: + if not params.print_diagnostics: scan_pessimistic_batches_for_oom( model=model, train_dl=train_dl,