TensorFlow 模型建立与训练

GPU 显存不够,改小 Batch Size 试试。