Skip to content

GitLab

Unverified Commit befbc73a authored by Benjamin Lefaudeux's avatar Benjamin Lefaudeux Committed by GitHub
Browse files

[CI] MNIST download fix (#581) 

* fixing given torchvision's change
parent 1fcbd624
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 18 deletions
benchmarks/datasets/mnist.py
View file @ befbc73a
...@@ -12,24 +12,7 @@ def setup_cached_mnist(): ...@@ -12,24 +12,7 @@ def setup_cached_mnist():
done, tentatives = False, 0 done, tentatives = False, 0
while not done and tentatives < 5: while not done and tentatives < 5:
# Monkey patch the resource URLs to work around a possible blacklist # Monkey patch the resource URLs to work around a possible blacklist
MNIST.resources = [ MNIST.mirrors = ["https://github.com/blefaudeux/mnist_dataset/raw/main/"] + MNIST.mirrors
(
"https://github.com/blefaudeux/mnist_dataset/raw/main/train-images-idx3-ubyte.gz",
"f68b3c2dcbeaaa9fbdd348bbdeb94873",
),
(
"https://github.com/blefaudeux/mnist_dataset/raw/main/train-labels-idx1-ubyte.gz",
"d53e105ee54ea40749a09fcbcd1e9432",
),
(
"https://github.com/blefaudeux/mnist_dataset/raw/main/t10k-images-idx3-ubyte.gz",
"9fb629c4189551a2d022fa330f9573f3",
),
(
"https://github.com/blefaudeux/mnist_dataset/raw/main/t10k-labels-idx1-ubyte.gz",
"ec29112dd5afa0611ce80d1b7f02629c",
),
]
# This will automatically skip the download if the dataset is already there, and check the checksum # This will automatically skip the download if the dataset is already there, and check the checksum
try: try:
......
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