Update spatial_transformer_tutorial.py (pytorch#1386)

* Update spatial_transformer_tutorial.py

* Update fgsm_tutorial.py

* Update beginner_source/fgsm_tutorial.py

Co-authored-by: Eli Uriegas <1700823+seemethere@users.noreply.github.com>

Co-authored-by: Eli Uriegas <1700823+seemethere@users.noreply.github.com>

Author: brianjo

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.
 # 
-

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