我定义了一个名为的自定义图层CapsuleLayer。实际模型已在单独的类中定义。我已将权重加载到实际模型中,并将模型保存在.h5文件中。但是,当我尝试使用加载模型时load_model(filepath),出现错误
CapsuleLayer
.h5
load_model(filepath)
ValueError:未知层:CapsuleLayer
加载保存的模型时,如何将自定义图层合并到模型中。
python keras keras-layer
keras ×1
keras-layer ×1
python ×1