TensorFlow 常用模块

我在带 GPU 的 Colab 环境里运行了一下你的代码,似乎没什么问题。我不知道是不是由于训练时间过短造成的(这个数据集比较大,训练需要花一点时间)。