download_dataset_from_hub.py 455 B

1234567891011
  1. def download_dataset():
  2. pass
  3. def download_dataset_from_ms(**kwargs):
  4. from modelscope.msdatasets import MsDataset
  5. dataset_name = kwargs.get("dataset_name", 'speech_asr/speech_asr_aishell1_trainsets')
  6. subset_name = kwargs.get("subset_name", 'default')
  7. split = kwargs.get("split", 'train')
  8. data_dump_dir = kwargs.get("data_dump_dir", None)
  9. ds = MsDataset.load(dataset_name=dataset_name, subset_name=subset_name, split=split, cache_dir=data_dump_dir)