[How do I get flair to recognize my custom embeddings class during model loading?]: about flair HOT 2 CLOSED

Hi @DonaldFeuz

I see you already figured out the from_params and to_params method. What is missing is the @register_embeddings decorator.

from flair.embeddings.base import register_embeddings

@register_embeddings
class MyEmbeddings(TokenEmbeddings):
   ....

that way, the lookup will know to also consider this embeddings class when loading embeddings.

Keep in mind, that the Moduel that defines the Embedding class needs to be loaded beforehands, e.g. you need to import it. I usually use the __init__.py to do that:

import my_project.my_embeddings

__all__ = ["my_embeddings"]
``

Then I can be sure that any file in `my_project` has the embeddings already registered and can load the model the correct way

from flair.

Thank you very much @helpmefindaname
My problem has been successfully solved thanks to your solution.

from flair.