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

Question

如何在训练后保存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（最近一次调用   最后）in（）   ----＆GT; 1 custom_vgg_model.load_weights（weights-Test-CNN.hdf5，by_name = False）

     

NameError：未定义名称'权重'

Answer 1

在第二个权重-Test-CNN.hdf5周围加上引号，或者您使用已定义的 fname ...

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