博客

keras自定义卷积层，出错ValueError: Unknown layer

  • Post author:
  • Post category:其他




出错原因：

要保存的model中包含了自定义的层（Custom Layer），导致加载模型的时候无法解析该Layer



解决办法

ner_model = load_model(model_path + 'my_model.h5', custom_objects={"CRF": CRF,"crf_loss": crf_loss, "crf_accuracy": crf_accuracy})



参考链接


参考


版权声明：本文为Student_PAN原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。