在使用keras时经常会使用到存储模型和加载模型。在存储时使用
model.save("model.h5")
如果此时存储的model中有自定义的loss或者metrics,如果使用model = keras.models.load_model("model.h5")
在加载模型的时候会报错:
ValueError: Unknown metric function:acc_top5
这是因为自定义的函数没有被保存,加载出来就会报错,解决方案为:
model = keras.models.load_model("model.h5", custom_objects={'acc_top5': acc_top5})
如果是自定义loss,则custom_objects={}
括号中做对应修改:
model = keras.models.load_model("model.h5", custom_objects={'loss_fun': loss_fun})