diff --git a/AudioLoader/speech/mls.py b/AudioLoader/speech/mls.py index d0e724b..9abad4f 100644 --- a/AudioLoader/speech/mls.py +++ b/AudioLoader/speech/mls.py @@ -18,7 +18,7 @@ major_version = int(split_version[0]) if major_version > 1: __TORCH_GTE_2_0 = True - from torchaudio.datasets.utils import _extract_zip as extract_archive + from torchaudio.datasets.utils import _extract_tar as extract_archive from torch.hub import download_url_to_file as download_url else: from torchaudio.datasets.utils import ( @@ -153,7 +153,7 @@ def __init__(self, print(f'{download_path+ext_archive} already exists, proceed to extraction...') else: print(f'downloading...') - download_url(url, root, hash_value=checksum, hash_type='md5') + download_url(url, download_path+ext_archive) if decision.lower()=='yes': print(f'extracting...') @@ -441,4 +441,4 @@ def _write_labels(self, audio_file, labels, root, Text, IPA): txt_output = label_id + '\t' + utterance with open(output_file, 'a') as f: - f.write(txt_output) \ No newline at end of file + f.write(txt_output)