fix the enh_s2t_task argument in espnet2/bin/st_inference.py

chintu619 · Apr 27, 2022 · 89a4d60 · 89a4d60
1 parent 40202fc
commit 89a4d60
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/espnet2/bin/st_inference.py b/espnet2/bin/st_inference.py
@@ -489,6 +489,12 @@ def get_parser():
         help="Pretrained model tag. If specify this option, *_train_config and "
         "*_file will be overwritten",
     )
+    group.add_argument(
+        "--enh_s2t_task",
+        type=str2bool,
+        default=False,
+        help="enhancement and asr joint model",
+    )
 
     group = parser.add_argument_group("Beam-search related")
     group.add_argument(