只保存和加载网络权重
若权重名后有.h5或.keras后缀,则保存为HDF5格式文件,否则默认为TensorFlow Checkpoint格式文件。
代码:
model.save_weights('./model/model_weights')
model.save_weights('./model/model_weights.h5')
权重加载
model.load_weights('./model/model_weights')
model.load_weights('./model/model_weights.h5')