MAIF / melusine

📧 Melusine: Use python to automatize your email processing workflow

Home Page:https://maif.github.io/melusine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tensorflow NotImplementedError error with Camembert

Kirire opened this issue · comments

Hi,

I think I have detected a bug when we want to save the model (save_nn_model / load_nn_model) in the melusine.models.train module with the pretrained Camembert hidden layer. I received a Python NotImplementedError error from Tensorflow library (from get_config). I think this is because TFCamembertModel is a Subclass Networks and isn't a Graph Networks. It seems there are answers in this topic.

Best regards

Hi,

Thank you so much for finding a bug! We'll take a look as quickly as possible and fix it.

Best regards

Hi @Kirire, we released Melusine 2.2.5 last week and it should fix this issue. The current transformers library does not allow to use the keras save function at its full potential. We implemented a work around. I will close this issue now.