tf.keras.models.load_model 加载模型报错记录

Unknown activation function:leaky_relu
解决办法是在tf.keras.models.load_model()括号里面加上

custom_objects = {“YYY”:XXX对应的函数或者对象}

例如报错 ValueError: Unknown activation function:leaky_relu 就加上

new_model=tf.keras.models.load_model("models/model.h5", custom_objects = {"leaky_relu":tf.nn.leaky_relu})