Skip to content

GitLab

"torchvision/vscode:/vscode.git/clone" did not exist on "7492fae4c2cd16fb2783dce7e7583d7245cfbe92"
Unverified Commit a485b8c6 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

simplify QMNIST download logic (#4562)

parent 05dcf50a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
torchvision/datasets/mnist.py
View file @ a485b8c6
...@@ -466,10 +466,7 @@ class QMNIST(MNIST): ...@@ -466,10 +466,7 @@ class QMNIST(MNIST):
split = self.resources[self.subsets[self.what]] split = self.resources[self.subsets[self.what]]
for url, md5 in split: for url, md5 in split:
filename = url.rpartition("/")[2] download_and_extract_archive(url, self.raw_folder, md5=md5)
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)
def __getitem__(self, index: int) -> Tuple[Any, Any]: def __getitem__(self, index: int) -> Tuple[Any, Any]:
# redefined to handle the compat flag # redefined to handle the compat flag
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment