代码和上一讲只是相差了保存权重和增加了预测功能 1.权重保存的路径和读取方法 checkpoint_save_path = "./checkpoint/mnist.ckpt" if os.path.exists(checkpoint_save_path+'.index'): print('---------------------load the model------------------') model.load_weights(checkpoint_save_path)