TransformersWsz / UMGF

Multi-modal Graph Fusion for Named Entity Recognition with Targeted Visual Guidance
65 stars 8 forks source link

Failed to reproduce the results #5

Open wangxinyu0922 opened 2 years ago

wangxinyu0922 commented 2 years ago

Thank you for your released codes. I tried to follow your instruction to train the model but I cannot reproduce your reported results in the paper. The F1 score on Twitter 15 and Twitter 17 datasets I got is shown as follows:

image image

I have tried to run the do_test command but the F1 score does not change significantly. Moreover, I have run the experiment over 5 times on each dataset but didn't find a significant difference in the score. Do you have any suggestions?

TransformersWsz commented 2 years ago

Thank you for your questions.

wangxinyu0922 commented 2 years ago

My CUDA version and driver version is identical to yours. I tried your checkpoint and the score is identical to yours (by the way, the model is saved in cuda:3. It is not good). However, I have no idea how to train such a model. Moreover, could you provide a model for twitter 17?

TransformersWsz commented 2 years ago

2022-01-08-15-37-31-image

Do you keep the same hyperparameters with mine?