首页 PaddleNLP 帖子详情
ERNIE finetune模型加载
收藏
快速回复
PaddleNLP 问答NLP 2702 5
ERNIE finetune模型加载
收藏
快速回复
PaddleNLP 问答NLP 2702 5

paddlehub的ERNIE通过finetune得到的模型下次使用时如何加载 

0
收藏
回复
全部评论(5)
时间顺序
AIStudio810258
#2 回复于2020-05

用预测脚本时,指定存储的模型路径

0
回复
j
jyslcg
#3 回复于2020-05

使用finetune下次预测时候加载模型的函数是哪一个呢

0
回复
无语凝咽白羊
#4 回复于2021-07
jyslcg #3
使用finetune下次预测时候加载模型的函数是哪一个呢

你好,我也遇到了这个问题,请问您解决了吗。

0
回复
无语凝咽白羊
#5 回复于2021-07
你好,我也遇到了这个问题,请问您解决了吗。

已解决:

我的checkpoint保存路径如下:

model:/home/nlp306/xgt/sentiment_analysis/checkpoint/model_checkpoint

tokenizer:/home/nlp306/xgt/sentiment_analysis/checkpoint/tokenizer_checkpoint

则想要在预测时加载这些参数只需如下操作(模型初始化时指定参数文件路径):

model = ppnlp.transformers.ErnieForSequenceClassification.from_pretrained(

'/home/nlp306/xgt/sentiment_analysis/checkpoint/model_checkpoint', num_classes=2)
tokenizer = ppnlp.transformers.ErnieTokenizer.from_pretrained(

'/home/nlp306/xgt/sentiment_analysis/checkpoint/tokenizer_checkpoint')

 

1
回复
深渊上的坑
#6 回复于2021-07
已解决: 我的checkpoint保存路径如下: model:/home/nlp306/xgt/sentiment_analysis/checkpoint/model_checkpoint tokenizer:/home/nlp306/xgt/sentiment_analysis/checkpoint/tokenizer_checkpoint 则想要在预测时加载这些参数只需如下操作(模型初始化时指定参数文件路径): model = ppnlp.transformers.ErnieForSequenceClassification.from_pretrained( '/home/nlp306/xgt/sentiment_analysis/checkpoint/model_checkpoint', num_classes=2) tokenizer = ppnlp.transformers.ErnieTokenizer.from_pretrained( '/home/nlp306/xgt/sentiment_analysis/checkpoint/tokenizer_checkpoint')  
展开

赞,可以指定路径

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户