Skip to content

GitLab

Unverified Commit cffaa589 authored by Qianli Scott Zhu's avatar Qianli Scott Zhu Committed by GitHub
Browse files

Update data model download location to temp dir. (#3899)

parent c79ae385
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/mnist/dataset.py
View file @ cffaa589
...@@ -20,6 +20,7 @@ from __future__ import print_function ...@@ -20,6 +20,7 @@ from __future__ import print_function
import gzip import gzip
import os import os
import shutil import shutil
import tempfile
import numpy as np import numpy as np
from six.moves import urllib from six.moves import urllib
...@@ -67,7 +68,7 @@ def download(directory, filename): ...@@ -67,7 +68,7 @@ def download(directory, filename):
tf.gfile.MakeDirs(directory) tf.gfile.MakeDirs(directory)
# CVDF mirror of http://yann.lecun.com/exdb/mnist/ # CVDF mirror of http://yann.lecun.com/exdb/mnist/
url = 'https://storage.googleapis.com/cvdf-datasets/mnist/' + filename + '.gz' url = 'https://storage.googleapis.com/cvdf-datasets/mnist/' + filename + '.gz'
zipped_filepath = filepath + '.gz' _, zipped_filepath = tempfile.mkstemp(suffix='.gz')
print('Downloading %s to %s' % (url, zipped_filepath)) print('Downloading %s to %s' % (url, zipped_filepath))
urllib.request.urlretrieve(url, zipped_filepath) urllib.request.urlretrieve(url, zipped_filepath)
with gzip.open(zipped_filepath, 'rb') as f_in, \ with gzip.open(zipped_filepath, 'rb') as f_in, \
......
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