在使用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})