Update spatial_transformer_tutorial.py (#1386)

brianjo · seemethere · web-flow · commit 7449ed3728fc · 2021-03-03T17:48:22.000-05:00
* Update spatial_transformer_tutorial.py

* Update fgsm_tutorial.py

* Update beginner_source/fgsm_tutorial.py

Co-authored-by: Eli Uriegas &lt;1700823+seemethere@users.noreply.github.com&gt;

Co-authored-by: Eli Uriegas &lt;1700823+seemethere@users.noreply.github.com&gt;
diff --git a/beginner_source/fgsm_tutorial.py b/beginner_source/fgsm_tutorial.py
@@ -99,6 +99,13 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
+# NOTE: This is a hack to get around "User-agent" limitations when downloading MNIST datasets
+#       see, https://github.com/pytorch/vision/issues/3497 for more information
+from six.moves import urllib
+opener = urllib.request.build_opener()
+opener.addheaders = [('User-agent', 'Mozilla/5.0')]
+urllib.request.install_opener(opener)
+
 
 ######################################################################
 # Implementation
@@ -413,4 +420,3 @@ def test( model, device, test_loader, epsilon ):
 # it differs from FGSM. Then, try to defend the model from your own
 # attacks.
 # 
-
diff --git a/intermediate_source/spatial_transformer_tutorial.py b/intermediate_source/spatial_transformer_tutorial.py
@@ -47,6 +47,11 @@
 # standard convolutional network augmented with a spatial transformer
 # network.
 
+from six.moves import urllib
+opener = urllib.request.build_opener()
+opener.addheaders = [('User-agent', 'Mozilla/5.0')]
+urllib.request.install_opener(opener)
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Training dataset
