From 6709bf1e6325166fcb989b1dbb03344d6b90b7f8 Mon Sep 17 00:00:00 2001
From: Nagendra Goel <nagendra.goel@gmail.com>
Date: Thu, 27 Oct 2022 22:23:32 -0400
Subject: [PATCH] Update train.py (#635)

Add the missing step to add the arguments to the parser.
---
 egs/librispeech/ASR/pruned_transducer_stateless3/train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py b/egs/librispeech/ASR/pruned_transducer_stateless3/train.py
index 723b03e15..a8fd527d5 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