mnist

Lightning-Universe · Mar 12, 2021 · 82b3835 · 82b3835
1 parent d20aa3c
commit 82b3835
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -0,0 +1,6 @@
+import urllib
+
+# TorchVision hotfix https://github.com/pytorch/vision/issues/1938
+opener = urllib.request.build_opener()
+opener.addheaders = [('User-agent', 'Mozilla/5.0')]
+urllib.request.install_opener(opener)