Keras调用model.load_weights时报错

报错:ValueError: Unable to load weights saved in HDF5 format into a subclassed Model which has not created its variables yet. Call the Model first, then load the weights.

模型框架使用Sequential()时不会出现如上错误。但本人使用了keras下Model类,继承该类实现模型的搭建。训练时保存网络权重参数,代码如下

self.model.save_weights("vehicular_DQN.h5")

加载参数代码如下

self.model.load_weights("vehicular_DQN.h5")

运行时出现上述报错,报错原因是以H5格式加载子类模型的参数时,需要提前建立模型,规定输入网络的shape。格式为model.build(input_shape =())。代码样例如下

self.model.build(input_shape =(None,82))
self.model.load_weights("vehicular_DQN.h5")

你可能感兴趣的:(keras,python,tensorflow)