tensorflow - 保存 keras 模型以便将来恢复训练的最佳方法是什么?

标签 tensorflow machine-learning deep-learning keras

我对此进行了很多搜索。在 keras 文档中,它说 model.save() 保存所有相关信息,即模型架构、权重、优化器状态……

stackoverflow 上的其他一些帖子提到保存权重并在将来加载它们以恢复训练,但答案说这是错误的,因为它没有保存优化器状态。 我使用回调来保存最佳模型基础 om 验证准确性,并且它只保存权重。

如果权重不足以恢复训练,为什么回调只会节省权重?只是为了在测试集上进行评估?

那么如何正确保存最佳模型呢?为什么回调不使用 model.save() 来存储所有信息?我怎样才能实现这个目标?

最佳答案

根据 Keras ModelCheckPointCallback 文档,它会根据 save_weights_only 标志保存整个模型或仅保存权重,默认情况下为 False

https://keras.io/callbacks/#modelcheckpoint

关于tensorflow - 保存 keras 模型以便将来恢复训练的最佳方法是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49336377/

相关文章:

java - Tensorflow java中图像的批量分类

python - 如何在两个张量之间进行元素卷积?

python - 构建用于牛眼皮疹识别的卷积神经网络

python - 我是给 cross_val_score() 整个数据集还是只提供训练集?

r - 从 R 中稀疏矩阵第 i 行的非零值中减去向量的值

python - 尝试训练二元分类 SGDClassifier 时出现位置参数错误

deep-learning - torch : GPU Memory Leak

python - Tensorflow 对象检测 API : how to create tfrecords with images not containing any labels (hard negatives)?

python - 在 TensorFlow 中梯度下降后更新权重

performance - 感知器学习算法的参数调整