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

出错原因:

要保存的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})

参考链接

参考

你可能感兴趣的:(tensorflow)