From 334a4710b0b758323da86a51b1827aaad550ee6c Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Thu, 7 Oct 2021 14:58:03 +0200 Subject: [PATCH] simplify QMNIST download logic --- torchvision/datasets/mnist.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index 49e20438874..5445730fb23 100644 --- a/torchvision/datasets/mnist.py +++ b/torchvision/datasets/mnist.py @@ -466,10 +466,7 @@ def download(self) -> None: split = self.resources[self.subsets[self.what]] for url, md5 in split: - filename = url.rpartition("/")[2] - file_path = os.path.join(self.raw_folder, filename) - if not os.path.isfile(file_path): - download_and_extract_archive(url, self.raw_folder, filename=filename, md5=md5) + download_and_extract_archive(url, self.raw_folder, md5=md5) def __getitem__(self, index: int) -> Tuple[Any, Any]: # redefined to handle the compat flag