commit | 49d0cf3bd1d755c4823e7321f95d617b690486cb | [log] [tgz] |
---|---|---|
author | Yong Tang <yong.tang.github@outlook.com> | Sun Jun 09 21:38:39 2019 +0000 |
committer | Yong Tang <yong.tang.github@outlook.com> | Thu Aug 01 17:08:33 2019 +0000 |
tree | ddaea776eff3f88b47fa6c694b73f49cef0ee867 | |
parent | a4a40577482817d34a55484cf1a37c81544e3716 [diff] |
Fix tf.keras.dataset.mnist/reuters np v1.16.3 This fix add `allow_pickle=True` to fix the numpy issue in 1.16.3. See https://github.com/keras-team/keras/commit/34231971fa47cb2477b357c1a368978de4128294 and https://github.com/keras-team/keras/pull/12714 for related changes in keras-team/keras. Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
diff --git a/tensorflow/python/keras/datasets/mnist.py b/tensorflow/python/keras/datasets/mnist.py index 7e012c3..bbcdbea 100644 --- a/tensorflow/python/keras/datasets/mnist.py +++ b/tensorflow/python/keras/datasets/mnist.py
@@ -48,7 +48,7 @@ origin=origin_folder + 'mnist.npz', file_hash= '731c5ac602752760c8e48fbffcf8c3b850d9dc2a2aedcf2cc48468fc17b673d1') - with np.load(path) as f: + with np.load(path, allow_pickle=True) as f: x_train, y_train = f['x_train'], f['y_train'] x_test, y_test = f['x_test'], f['y_test']