grammarly / gector

Official implementation of the papers "GECToR – Grammatical Error Correction: Tag, Not Rewrite" (BEA-20) and "Text Simplification by Tagging" (BEA-21)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Loading TST-FINAL Model

yassinekdi opened this issue · comments

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

Hi @yassinekdi
Did you find a solution?