diff --git a/examples/emnist/download.py b/examples/emnist/download.py index 41cf9ea..78d2266 100644 --- a/examples/emnist/download.py +++ b/examples/emnist/download.py @@ -1,3 +1,4 @@ +import torchvision from torchvision.datasets.utils import download_url import os @@ -6,8 +7,14 @@ url = 'https://biometrics.nist.gov/cs_links/EMNIST/gzip.zip' md5 = "58c8d27c78d21e728a6bc7b3cc06412e" +version_numbers = list(map(int, torchvision.__version__.split('+')[0].split('.'))) +if version_numbers[0] == 0 and version_numbers[1] < 10: + filename = "emnist.zip" +else: + filename = None + os.makedirs(raw_folder, exist_ok=True) # download files print('Downloading zip archive') -download_url(url, root=raw_folder, filename="emnist.zip", md5=md5) +download_url(url, root=raw_folder, filename=filename, md5=md5)