保存在 keras 中训练的 VGG-16 模型的权重答案

【问题标题】：Saving the weights of VGG-16 model trained in keras保存在 keras 中训练的 VGG-16 模型的权重
【发布时间】：2018-08-03 18:20:03
【问题描述】：

VGG-16模型训练后如何保存权重？如何将保存的权重加载到模型中？

我试过这个：

fname = "weights-Test-CNN.hdf5"

custom_vgg_model.save_weights(fname,overwrite=True)
custom_vgg_model.load_weights(weights-Test-CNN.hdf5, by_name=False)

我收到以下错误：

NameError Traceback（最近调用最后）在（） ----> 1 custom_vgg_model.load_weights(weights-Test-CNN.hdf5, by_name=False)

NameError: 名称“权重”未定义

【问题讨论】：

Save and load weights in keras的可能重复
你只是忘记了 weights-Test-CNN.hdf5 周围的引号...custom_vgg_model.load_weights("weights-Test-CNN.hdf5", by_name=False)

标签： keras deep-learning vgg-net

【解决方案1】：

用引号将第二个 weights-Test-CNN.hdf5 括起来，或者按照您的定义使用 fname...

custom_vgg_model.load_weights("weights-Test-CNN.hdf5", by_name=False)

【讨论】：