diff --git a/examples/asr/asr_ctc/speech_to_text_ctc.py b/examples/asr/asr_ctc/speech_to_text_ctc.py index ebb91221c41f..788e1e1d6d4b 100644 --- a/examples/asr/asr_ctc/speech_to_text_ctc.py +++ b/examples/asr/asr_ctc/speech_to_text_ctc.py @@ -90,15 +90,8 @@ def main(cfg): trainer.fit(asr_model) if hasattr(cfg.model, 'test_ds') and cfg.model.test_ds.manifest_filepath is not None: - gpu = 1 if cfg.trainer.gpus != 0 else 0 - test_trainer = pl.Trainer( - gpus=gpu, - precision=trainer.precision, - amp_level=trainer.accelerator_connector.amp_level, - amp_backend=cfg.trainer.get("amp_backend", "native"), - ) - if asr_model.prepare_test(test_trainer): - test_trainer.test(asr_model) + if asr_model.prepare_test(trainer): + trainer.test(asr_model) if __name__ == '__main__': diff --git a/examples/asr/asr_ctc/speech_to_text_ctc_bpe.py b/examples/asr/asr_ctc/speech_to_text_ctc_bpe.py index 9114be640c53..51b2e9444bc8 100644 --- a/examples/asr/asr_ctc/speech_to_text_ctc_bpe.py +++ b/examples/asr/asr_ctc/speech_to_text_ctc_bpe.py @@ -86,15 +86,8 @@ def main(cfg): trainer.fit(asr_model) if hasattr(cfg.model, 'test_ds') and cfg.model.test_ds.manifest_filepath is not None: - gpu = 1 if cfg.trainer.gpus != 0 else 0 - test_trainer = pl.Trainer( - gpus=gpu, - precision=trainer.precision, - amp_level=trainer.accelerator_connector.amp_level, - amp_backend=cfg.trainer.get("amp_backend", "native"), - ) - if asr_model.prepare_test(test_trainer): - test_trainer.test(asr_model) + if asr_model.prepare_test(trainer): + trainer.test(asr_model) if __name__ == '__main__': diff --git a/examples/asr/asr_transducer/speech_to_text_rnnt.py b/examples/asr/asr_transducer/speech_to_text_rnnt.py index b96d7c254ab1..a77a58da97fd 100644 --- a/examples/asr/asr_transducer/speech_to_text_rnnt.py +++ b/examples/asr/asr_transducer/speech_to_text_rnnt.py @@ -89,15 +89,8 @@ def main(cfg): trainer.fit(asr_model) if hasattr(cfg.model, 'test_ds') and cfg.model.test_ds.manifest_filepath is not None: - gpu = 1 if cfg.trainer.gpus != 0 else 0 - test_trainer = pl.Trainer( - gpus=gpu, - precision=trainer.precision, - amp_level=trainer.accelerator_connector.amp_level, - amp_backend=cfg.trainer.get("amp_backend", "native"), - ) - if asr_model.prepare_test(test_trainer): - test_trainer.test(asr_model) + if asr_model.prepare_test(trainer): + trainer.test(asr_model) if __name__ == '__main__': diff --git a/examples/asr/asr_transducer/speech_to_text_rnnt_bpe.py b/examples/asr/asr_transducer/speech_to_text_rnnt_bpe.py index 91e41deaf05a..da93dbee410b 100644 --- a/examples/asr/asr_transducer/speech_to_text_rnnt_bpe.py +++ b/examples/asr/asr_transducer/speech_to_text_rnnt_bpe.py @@ -81,15 +81,8 @@ def main(cfg): trainer.fit(asr_model) if hasattr(cfg.model, 'test_ds') and cfg.model.test_ds.manifest_filepath is not None: - gpu = 1 if cfg.trainer.gpus != 0 else 0 - test_trainer = pl.Trainer( - gpus=gpu, - precision=trainer.precision, - amp_level=trainer.accelerator_connector.amp_level, - amp_backend=cfg.trainer.get("amp_backend", "native"), - ) - if asr_model.prepare_test(test_trainer): - test_trainer.test(asr_model) + if asr_model.prepare_test(trainer): + trainer.test(asr_model) if __name__ == '__main__': diff --git a/examples/asr/speech_classification/speech_to_label.py b/examples/asr/speech_classification/speech_to_label.py index 0b68484ba49a..8a83d41543e0 100644 --- a/examples/asr/speech_classification/speech_to_label.py +++ b/examples/asr/speech_classification/speech_to_label.py @@ -136,8 +136,6 @@ def main(cfg): trainer.fit(asr_model) if hasattr(cfg.model, 'test_ds') and cfg.model.test_ds.manifest_filepath is not None: - gpu = 1 if cfg.trainer.gpus != 0 else 0 - trainer = pl.Trainer(gpus=gpu) if asr_model.prepare_test(trainer): trainer.test(asr_model)