keras mnist.load_data() 速度超慢，一段时间后会抛出错误

这是我使用的完整代码

import os
import numpy as np
import matplotlib.pyplot as plt
from tqdm import tqdm


from keras.layers import Input
from keras.models import Model, Sequential
from keras.layers.core import Dense, Dropout
from keras.layers.advanced_activations import LeakyReLU
from keras.datasets import mnist
from keras.optimizers import Adam
from keras import initializers

os.environ["KERAS_BACKEND"] = "tensorflow"

np.random.seed(10)

random_dim = 100

def load_mnist_data():
    # load the data
    (x_train, y_train), (x_test, y_test) = mnist.load_data()
    # normalize our inputs to be in the range[-1, 1]
    x_train = (x_train.astype(np.float32) - 127.5)/127.5
    print(x_train.shape)
    # convert x_train with a shape of (60000, 28, 28) to (60000, 784) so we have
    # 784 columns per row
    x_train = x_train.reshape(60000, 784)
    return (x_train, y_train, x_test, y_test)


load_mnist_data()

大约需要 30 分钟才能下载一半，一段时间后（大约 5000000/11490434 和 6000000/11490434）它崩溃并抛出此错误：

ConnectionResetError: [WinError 10054] 存在远程连接

（翻译）

现有连接已被远程主机关闭

是什么原因造成的？用 keras 下载 mnist 数据需要一个小时，而我从 TensorFlow 下载它只需要几分钟，这正常吗？

我在评论中写了如何解决问题，但我想让它更清楚。

仅当程序从空闲状态启动并且这是第一次从 keras 导入 mnist 数据时，才会出现此问题。

简单的解决方案是简单地从命令提示符启动你的程序，看起来空闲无法处理打印那么多东西，并且在空闲时打印一些东西实际上会在很短的时间内停止你的程序，这可以解释荒谬的时间量下载它。

TLDR：从命令提示符启动程序