仅保存和加载权重
save_weights method
Model.save_weights(filepath, overwrite=True)
保存所有层权重到.weights.h5文件.
参数:
filepath: str 或 pathlib.Path 对象.
保存模型的路径.必须以.weights.h5结尾.
overwrite: 是否应该覆盖目标位置上的任何现有模型,
或者通过交互式提示询问用户.
load_weights method
Model.load_weights(filepath, skip_mismatch=False, **kwargs)
从通过save_weights()保存的文件中加载权重.
权重根据网络的拓扑结构进行加载.这意味着架构应该与保存权重时的架构相同.请注意,没有权重的层在拓扑排序中不被考虑,因此添加或删除这些层是可以的,只要它们没有权重.
部分权重加载
如果你修改了模型,例如通过添加一个新层（带有权重）或通过改变某层权重的形状,你可以选择忽略错误并继续加载,通过设置skip_mismatch=True.在这种情况下,任何权重不匹配的层将被跳过.每个被跳过的层都会显示一个警告.
参数:
filepath: 字符串,要加载的权重文件的路径.
它可以是.weights.h5文件
或一个旧版的.h5权重文件.
skip_mismatch: 布尔值,是否跳过加载权重数量不匹配或权重形状不匹配的层.