diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py b/egs/librispeech/ASR/pruned_transducer_stateless3/train.py index 723b03e156..a8fd527d57 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless3/train.py @@ -327,7 +327,9 @@ def get_parser(): default=0.5, help="The probability to select a batch from the GigaSpeech dataset", ) - + + add_model_arguments(parser) + return parser