代码和上一讲只是相差了保存权重和增加了预测功能 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) 2.callback权重填写函数 cp_callback = tf.keras.callbacks.