Failed to load pretrained models

declare-lab / MELD

MELD: A Multimodal Multi-Party Dataset for Emotion Recognition in Conversation

GNU General Public License v3.0

788 stars 200 forks source link

Failed to load pretrained models #7

Closed KuzmaNg closed 5 years ago

KuzmaNg commented 5 years ago

Hi, I tried to reload the pretrained models, but failed. I assume it is a keras version problem. Would you like to describe the running enviroment? Sorry, I cannot find it in Readme. Thanks.

soujanyaporia commented 5 years ago

What problem did you encounter? We used Keras > 2.0.2 and latest tf as a backend.

KuzmaNg commented 5 years ago

Thanks. @soujanyaporia The problem is, unexpected keyword argument is found when reloading the pretrained model. I changed the keras to 2.1.5, and solved the problem. But I still cannot reproduce the result reported in paper MELD. Keep trying. : )

soujanyaporia commented 5 years ago

I guess you are not using class weights. Please do use class weights to reproduce the results. I think we will release it in next update.

KuzmaNg commented 5 years ago

Thank you for the hint. Aha