使用tenseoflow自定义一个网络,为类,训练好之后,加载模型,出现ValueError: Unknown layer: ExternalAttention。下面分别是定义的类和报错。
自定义的类:
报错:
在网上搜寻了方法,直接添加一个参数
model = load_model('lstm2.h5', custom_objects={'ExternalAttention': ExternalAttention})
结果出现下列错误:
意思应该是出现了多个值,也出现了名字,应该是多了名字这个参数,回到自定义类,将名字参数删除即可。
删除前截图:第五行有个name参数
删除后截图:
成功运行