想加载自己定义的数据集非test.txt去做预测，怎么保存模型、加载模型呢？

LeeSureman / Flat-Lattice-Transformer

code for ACL 2020 paper: FLAT: Chinese NER Using Flat-Lattice Transformer

1k stars 178 forks source link

Open s1162276945 opened 3 years ago

s1162276945 commented 3 years ago

想加载自己定义的数据集非test.txt，怎么保存模型、加载模型呢，预测函数感觉和fastNLP中文文档不太一样，不太会写。

LeeSureman commented 3 years ago

可以在trainer.train()后面加上对于model的保存操作

s1162276945 commented 3 years ago

可以在trainer.train()后面加上对于model的保存操作

加上了，但是加载的时候出错了，可以请教一下你么

s1162276945 commented 3 years ago

可以在trainer.train()后面加上对于model的保存操作

加上了，但是加载的时候出错了，可以请教一下你么

已经发邮件给你了，还请帮忙看一下，谢谢。

LeeSureman commented 3 years ago

感觉像是dataset出了问题？

zelin-x commented 3 years ago

可以在trainer.train()后面加上对于model的保存操作

加上了，但是加载的时候出错了，可以请教一下你么

请问您解决这个问题了吗

zelin-x commented 3 years ago

在trainer.train()后保存trainer.model, 可以使用FastNLP的Tester进行测试，数据处理需要与原来的训练数据处理一致

Shajiu commented 2 years ago

在trainer.train()后保存trainer.model, 可以使用FastNLP的Tester进行测试，数据处理需要与原来的训练数据处理一致

你好！请问具体怎么操作么？方便交流一下么？我这块实在没法入手。

Shajiu commented 2 years ago

可以在trainer.train()后面加上对于model的保存操作

加上了，但是加载的时候出错了，可以请教一下你么

请问您解决这个问题了吗

您好！请您您具体怎么处理的呀！感谢大佬~

CookieY1 commented 1 year ago

请问如何处理数据保证与原来的训练数据处理一致