仅权重保存和加载

[source]

save_weights 方法

Model.save_weights(filepath, overwrite=True)

将所有层的权重保存到 .weights.h5 文件。

参数

• filepath: str 或 pathlib.Path 对象。模型保存路径。必须以 .weights.h5 结尾。
• overwrite: 是否应覆盖目标位置的任何现有模型，或者通过交互式提示询问用户。

[source]

load_weights 方法

Model.load_weights(filepath, skip_mismatch=False, **kwargs)

从通过 save_weights() 保存的文件加载权重。

权重根据网络的拓扑结构加载。这意味着架构应与保存权重时相同。请注意，拓扑排序中不考虑没有权重的层，因此添加或删除层是可以的，只要它们没有权重。

部分权重加载

如果您修改了模型，例如通过添加新层（带有权重）或更改层权重的形状，您可以选择忽略错误并通过设置 skip_mismatch=True 继续加载。在这种情况下，任何权重不匹配的层都将被跳过。将为每个跳过的层显示警告。

参数

• filepath: 字符串，要加载的权重文件的路径。它可以是 .weights.h5 文件或旧版 .h5 权重文件。
• skip_mismatch: 布尔值，是否跳过加载权重数量不匹配或权重形状不匹配的层。