search

grammarly / gector

Official implementation of the papers "GECToR – Grammatical Error Correction: Tag, Not Rewrite" (BEA-20) and "Text Simplification by Tagging" (BEA-21)
Apache License 2.0
900 stars 214 forks source link

Loading TST-FINAL Model #147

Closed yassinekdi closed 2 years ago

yassinekdi commented 2 years ago

Thank you for the amazing work with the Text Simplification! I'm just stuck at loading the weights (in the th model). I could just get the dictionary ( torch.load(yourmodel.th) ) but I couldn't use the loaded weights in any model. I tried use huggingface models (Roberta) but it's not working. I read you used AllenNLP in your papers. I tried AllenNLP models but still couldn't load it. Any Idea ? Thank you :)

skurzhanskyi commented 2 years ago

Hi @yassinekdi Did you find a solution?

yassinekdi commented 2 years ago

Hi,Actually I just ended up using the python command (python predict.py etc.). Since I just wanted to take a look on the results that was enough :)  Le jeudi 30 décembre 2021, 12:47:11 UTC+1, Alex Skurzhanskyi @.***> a écrit :

Hi @yassinekdi Did you find a solution?

— Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android. You are receiving this because you were mentioned.Message ID: @.***>