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 版权协议,转载请附上原文出处链接和本声明。