游雁
2023-02-08 6d8528fa9f0ca9261718ef0afec67a4acb48c6d7
funasr/utils/modelscope_param.py
@@ -1,4 +1,3 @@
class modelscope_args():
   def __init__(self,
               task: str = "",
@@ -9,7 +8,14 @@
               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
@@ -19,7 +25,11 @@
      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