如何不在 Tensorflow Keras 中保存模型优化器？

Question

我正在阅读有关save_weights保存加载的官方教程，似乎无论我使用了save还是save_weights方法，无论如何都会保存优化器参数。 如何仅保存模型的权重？

Answer 1

在 Keras 中，要保存模型权重，请执行以下操作：

model.save_weights('my_model_weights.h5')

加载模型权重：

model.load_weights('my_model_weights.h5')

另请参阅此处有关按图层名称保存/加载权重的其他示例。

Answer 2

model.save('./savedmodel.h5', save_format='h5', include_optimizer=False)

如果save_format='tf' ，无论是include_optimizer=False还是True ，我都试过它没用。