Fix support for torch 2.0

Author: marvinborner

AudioLoader/speech/speechcommands.py

@@ -5,6 +5,7 @@
 from torch import Tensor
 import os
 import tqdm
+import torch
 __TORCH_GTE_2_0 = False
 split_version = torch.__version__.split(".")
 major_version = int(split_version[0])
@@ -282,4 +283,4 @@ def __getitem__(self, n: int) -> Tuple[Any, Any]:
         return waveform, label
 
     def __len__(self) -> int:
-        return len(self._data)       
+        return len(self._data)       

AudioLoader/speech/timit.py

@@ -13,10 +13,18 @@
 import warnings
 from distutils.dir_util import copy_tree
 from torchaudio.compliance import kaldi # for downsampling
-from torchaudio.datasets.utils import (
-    download_url,
-    extract_archive,
-)
+__TORCH_GTE_2_0 = False
+split_version = torch.__version__.split(".")
+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 torch.hub import download_url_to_file as download_url
+else:
+    from torchaudio.datasets.utils import (
+        download_url,
+        extract_archive,
+    )
 import hashlib
 import torch.nn.functional as F            
 from AudioLoader.music.utils import check_md5
@@ -183,4 +191,4 @@ def available_groups(self, groups):
         if groups=='all':
             return [f'DR{i+1}' for i in range(9)] # select all dialect regions
         elif isinstance(groups, list):
-            return [f'DR{i}' for i in groups]   
+            return [f'DR{i}' for i in groups]