modelscope_param.py 709 B

12345678910111213141516171819202122232425
  1. class modelscope_args():
  2. def __init__(self,
  3. task: str = "",
  4. model: str = "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
  5. data_path: str = None,
  6. output_dir: str = None,
  7. model_revision: str = None,
  8. dataset_type: str = "small",
  9. batch_bins: int = 2000,
  10. max_epoch: int = None,
  11. lr: float = None,
  12. ):
  13. self.task = task
  14. self.model = model
  15. self.data_path = data_path
  16. self.output_dir = output_dir
  17. self.model_revision = model_revision
  18. self.dataset_type = dataset_type
  19. self.batch_bins = batch_bins
  20. self.max_epoch = max_epoch
  21. self.lr = lr