模型的保存

Question

老师tf.saved_model.save(model, "./keras_saved_graph")和model.save_weights(os.path.join(logdir, "fashion_mnist_weights_2.h5"))保存下来是最后一次参数更新的模型还是在验证集上效果最好的呢

Answer 1

正十七 2020-01-02 22:24:29

同学你好，题目中的这两个方法保存下来的模型是最新的参数状态，如果想要保存最好的，需要使用ModelCheckPointCallback:

1 2 3

keras.callbacks.ModelCheckpoint(output_model_file,                                     save_best_only = True,                                     save_weights_only = True),

这里的save_best_only，就是用来保存最好的模型。