From 3d9f094e9652d4b84894c6fd4eae39a4a753b0f0 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 16 五月 2023 23:48:00 +0800
Subject: [PATCH] train

---
 funasr/bin/train.py |   14 +++++++++++++-
 1 files changed, 13 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 8436dd5..53e5bde 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -334,13 +334,25 @@
         default="validation",
         help="dev dataset",
     )
-
+    parser.add_argument(
+        "--speed_perturb",
+        type=float,
+        nargs="+",
+        default=None,
+        help="speed perturb",
+    )
     parser.add_argument(
         "--use_preprocessor",
         type=str2bool,
         default=True,
         help="Apply preprocessing to data or not",
     )
+    parser.add_argument(
+        "--embed_path",
+        type=str,
+        default=None,
+        help="for model which requires embeds",
+    )
 
     # optimization related
     parser.add_argument(

--
Gitblit v1.9.1