从keras Python加载mnist数据集时出现连接错误

Question

我正在尝试使用Python的keras库加载mnist数据集：

from keras.datasets import mnist

(xt,yt),(xv,yv)=mnist.load_data()

数据集开始正常下载，但过一会儿出现以下错误：

  

ConnectionResetError：[WinError 10054]现有连接被远程主机强行关闭

我尝试从另一个来源下载相同的数据集：

from tensorflow.keras import datasets
(xt,yt),(xv,yv)=datasets.mnist.load_data()

但我仍然遇到相同的错误。

我也尝试了以下方法：

mnist=input_data.read_data_sets("MNIST_data/",one_hot=False)
x_train=mnist.train.images

y_train=mnist.train.labels

x_test=mnist.test.images

y_test=mnist.test.labels

数据加载成功，但x_train和x_test已作为2D数组加载（已经重塑）。我实际上想要的是数据的原始3D版本（在重塑之前），但似乎无法成功加载。