CyberZHG/keras-multi-head

How to load a keras_multi_head model?

Closed this issue · 2 comments

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.

custom_objects is required.