保存在keras中训练的VGG-16模型的重量

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回溯（最近的调用） 最后）在（） ---->1个自定义vgg模型。加载权重（权重-Test-CNN.hdf5，按名称=False）

NameError:未定义名称“权重”

---

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

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

---

可能是你忘记了重量测试CNN.hdf5周围的引号<代码>自定义vgg模型。加载权重（“权重测试CNN.hdf5”，由\u name=False）