模型的保存

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:

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

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