我正在阅读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')
另请参阅此处关于按层名称保存/加载权重的其他示例。