Update train.py (#635)

Add the missing step to add the arguments to the parser.
k2-fsa · Oct 28, 2022 · 6709bf1 · 6709bf1
1 parent 499ac24
commit 6709bf1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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