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



我正在阅读Keras中保存负载的官方教程,无论我使用的是save还是save_weights方法,优化器参数都会被保存。如何只保存模型的权重?

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

如果是save_format='tf',无论是include_optimizer=False还是True,都没有用。

在Keras中,要保存模型权重,请执行:

model.save_weights('my_model_weights.h5')

加载模型权重:

model.load_weights('my_model_weights.h5')

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

最新更新