模型的保存

来源：9-3 Keras模型转化为SavedModel

wxz123

2019-12-29

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

写回答

1回答

正十七

2020-01-02

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

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

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

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程

相似问题

Transformer模型怎么保存

回答 2

保存模型之后再次运行出错

回答 1

ModelCheckpoint保存的模型与tf.saved_model.save区别

回答 1

新定义的mode = WideDeepModel() 如何保存训练好的模型

回答 1

请问ModelCheckpoint保存的模型应该如何正确使用

回答 1

打开慕课网App查看更多内容