From 6d8528fa9f0ca9261718ef0afec67a4acb48c6d7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 08 二月 2023 19:58:16 +0800
Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add

---
 funasr/utils/modelscope_param.py |   58 ++++++++++++++++++++++++++++++++++------------------------
 1 files changed, 34 insertions(+), 24 deletions(-)

diff --git a/funasr/utils/modelscope_param.py b/funasr/utils/modelscope_param.py
index 5d6bffb..9ff196a 100644
--- a/funasr/utils/modelscope_param.py
+++ b/funasr/utils/modelscope_param.py
@@ -1,25 +1,35 @@
-
 class modelscope_args():
-	def __init__(self,
-	            task: str = "",
-	            model: str = "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
-	            data_path: str = None,
-	            output_dir: str = None,
-	            model_revision: str = None,
-	            dataset_type: str = "small",
-	            batch_bins: int = 2000,
-	            max_epoch: int = None,
-	            lr: float = None,
-	            ):
-		self.task = task
-		self.model = model
-		self.data_path = data_path
-		self.output_dir = output_dir
-		self.model_revision = model_revision
-		self.dataset_type = dataset_type
-		self.batch_bins = batch_bins
-		self.max_epoch = max_epoch
-		self.lr = lr
-		
-		
-		
\ No newline at end of file
+    def __init__(self,
+                 task: str = "",
+                 model: str = "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
+                 data_path: str = None,
+                 output_dir: str = None,
+                 model_revision: str = None,
+                 dataset_type: str = "small",
+                 batch_bins: int = 2000,
+                 max_epoch: int = None,
+                 accum_grad: int = None,
+                 keep_nbest_models: int = None,
+                 optim: str = None,
+                 lr: float = None,
+                 scheduler: str = None,
+                 scheduler_conf: dict = None,
+                 specaug: str = None,
+                 specaug_conf: dict = None,
+                 ):
+        self.task = task
+        self.model = model
+        self.data_path = data_path
+        self.output_dir = output_dir
+        self.model_revision = model_revision
+        self.dataset_type = dataset_type
+        self.batch_bins = batch_bins
+        self.max_epoch = max_epoch
+        self.accum_grad = accum_grad
+        self.keep_nbest_models = keep_nbest_models
+        self.optim = optim
+        self.lr = lr
+        self.scheduler = scheduler
+        self.scheduler_conf = scheduler_conf
+        self.specaug = specaug
+        self.specaug_conf = specaug_conf

--
Gitblit v1.9.1