fix #472

fangwei123456 · Dec 6, 2023 · cb1cee0 · cb1cee0
1 parent b70d5a9
commit cb1cee0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spikingjelly/datasets/speechcommands.py b/spikingjelly/datasets/speechcommands.py
@@ -6,7 +6,7 @@
 import torchaudio
 from torch.utils.data import Dataset
 from torch import Tensor
-from torchaudio.datasets.utils import (
+from torchvision.datasets.utils import (
     download_url,
     extract_archive
 )
@@ -122,7 +122,7 @@ def __init__(self,
             if not os.path.isdir(self._path):
                 if not os.path.isfile(archive):
                     checksum = _CHECKSUMS.get(url, None)
-                    download_url(url, root, hash_value=checksum, hash_type="md5")
+                    download_url(url, root, md5=checksum)
                 extract_archive(archive, self._path)
         elif not os.path.isdir(self._path):
             raise FileNotFoundError("Audio data not found. Please specify \"download=True\" and try again.")