ds2 model_type more info

deepspeech/exps/deepspeech2/bin/export.py

@@ -33,10 +33,9 @@ if __name__ == "__main__":
     # save jit model to 
     parser.add_argument(
         "--export_path", type=str, help="path of the jit model to save")
-    parser.add_argument("--model_type")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
     args = parser.parse_args()
-    if args.model_type is None:
-        args.model_type = 'offline'
     print("model_type:{}".format(args.model_type))
     print_arguments(args)
 

deepspeech/exps/deepspeech2/bin/test.py

@@ -30,14 +30,13 @@ def main(config, args):
 
 if __name__ == "__main__":
     parser = default_argument_parser()
-    parser.add_argument("--model_type")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
     # save asr result to 
     parser.add_argument(
         "--result_file", type=str, help="path of save the asr result")
     args = parser.parse_args()
     print_arguments(args, globals())
-    if args.model_type is None:
-        args.model_type = 'offline'
     print("model_type:{}".format(args.model_type))
 
     # https://yaml.org/type/float.html

deepspeech/exps/deepspeech2/bin/test_export.py

@@ -36,11 +36,10 @@ if __name__ == "__main__":
     #load jit model from
     parser.add_argument(
         "--export_path", type=str, help="path of the jit model to save")
-    parser.add_argument("--model_type")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
     args = parser.parse_args()
     print_arguments(args, globals())
-    if args.model_type is None:
-        args.model_type = 'offline'
     print("model_type:{}".format(args.model_type))
 
     # https://yaml.org/type/float.html

deepspeech/exps/deepspeech2/bin/test_hub.py

@@ -177,15 +177,14 @@ def main(config, args):
 
 if __name__ == "__main__":
     parser = default_argument_parser()
-    parser.add_argument("--model_type")
-    parser.add_argument("--audio_file")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
+    parser.add_argument("--audio_file", type=str, help='audio file path.')
     # save asr result to
     parser.add_argument(
         "--result_file", type=str, help="path of save the asr result")
     args = parser.parse_args()
     print_arguments(args, globals())
-    if args.model_type is None:
-        args.model_type = 'offline'
     if not os.path.isfile(args.audio_file):
         print("Please input the audio file path")
         sys.exit(-1)

deepspeech/exps/deepspeech2/bin/train.py

@@ -35,10 +35,9 @@ def main(config, args):
 
 if __name__ == "__main__":
     parser = default_argument_parser()
-    parser.add_argument("--model_type")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
     args = parser.parse_args()
-    if args.model_type is None:
-        args.model_type = 'offline'
     print("model_type:{}".format(args.model_type))
     print_arguments(args, globals())
 

examples/1xt2x/src_deepspeech2x/bin/test.py

@@ -31,14 +31,13 @@ def main(config, args):
 
 if __name__ == "__main__":
     parser = default_argument_parser()
-    parser.add_argument("--model_type")
+    parser.add_argument(
+        "--model_type", type=str, default='offline', help='offline/online')
     # save asr result to
     parser.add_argument(
         "--result_file", type=str, help="path of save the asr result")
     args = parser.parse_args()
     print_arguments(args, globals())
-    if args.model_type is None:
-        args.model_type = 'offline'
     print("model_type:{}".format(args.model_type))
 
     # https://yaml.org/type/float.html