keras调用load_model时报错TypeError: __init__() missing 1 required positional argument: ‘output_dim‘
自己编写的Self-Attention层, 将Self-Attention层用于训练,训练模型保存h5文件,评估模型的时候,调用load_model出了问题,报错信息:
ValueError: Unknown layer: AttentionSelf
从报错内容可知,是keras没认出自定义的Self-Attention层,于是按照网上的方法,给load_model添加了一个参数custom_objects={'AttentionSelf':AttentionSelf}
from keras.mod
原创
2022-01-11 10:07:22 ·
2975 阅读 ·
0 评论