declare-lab/MELD

Failed to load pretrained models

KuzmaNg opened this issue · 4 comments

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.

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

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. : )

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.

Thank you for the hint. Aha