Skip to content

GitLab

Unverified Commit 475a57d8 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

mnist downlaod workaround (#2122)

parent 444d2d7b
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
examples/trials/mnist-pytorch/mnist.py
View file @ 475a57d8
...@@ -15,6 +15,13 @@ import torch.nn.functional as F ...@@ -15,6 +15,13 @@ import torch.nn.functional as F
import torch.optim as optim import torch.optim as optim
from torchvision import datasets, transforms from torchvision import datasets, transforms
# Temporary patch this example until the MNIST dataset download issue get resolved
# https://github.com/pytorch/vision/issues/1938
import urllib
opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
urllib.request.install_opener(opener)
logger = logging.getLogger('mnist_AutoML') logger = logging.getLogger('mnist_AutoML')
......
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