mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

download mnist example data w/ six.moves.request

Browse Source 
fixes #28
This commit is contained in:
Matthew Johnson 2018-12-09 10:54:37 -08:00
parent 269de6010d
commit 38c7a07248
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/datasets.py
View File

@ -23,7 +23,7 @@ import gzip
import os
from os import path
import struct
import urllib2
from six.moves.urllib.request import urlretrieve
import numpy as np
@ -37,9 +37,8 @@ def _download(url, filename):
os.makedirs(_DATA)
out_file = path.join(_DATA, filename)
if not path.isfile(out_file):
with open(out_file, "wb") as f:
f.write(urllib2.urlopen(url, out_file).read())
print("downloaded {} to {}".format(url, _DATA))
urlretrieve(url, out_file)
print("downloaded {} to {}".format(url, _DATA))
def _partial_flatten(x):