How to load a keras_multi_head model?
Closed this issue · 2 comments
lxrswdd commented
I have trained a model using MultiHead layer.
When I tried to load it, it raises an error as ValueError: Unknown layer: MultiHead
.
I guess I have to add a custom_objects
but I am not sure what it is .
stale commented
Is this still relevant? If so, what is blocking it? Is there anything you can do to help move it forward?
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs.
CyberZHG commented
custom_objects
is required.