赞
踩
目录
利用自己的数据训练/微调模型
参考PaddleNLP SKEP情感分析训练示例,完成数据准备和模型训练。
指定了模型的缓存目录
cache_dir
,从远程服务器加载预训练模型时,PaddleNLP会首先将模型文件下载到这个缓存目录,以便后续可以更快地加载模型,而无需每次都从远程服务器下载。
- from paddlenlp.transformers import SkepForSequenceClassification, SkepTokenizer
-
- # 指定模型名称和缓存目录
- model_name = "skep_ernie_1.0_large_ch" # 预训练模型的名称
- cache_dir = "./paddlenlp/models" # 缓存目录,用于存储下载的模型文件
-
- # num_classes: 两类,0和1,表示消极和积极
- # 从指定的预训练模型名称或路径加载模型,并设置分类任务的类别数量
- # cache_dir 参数指定
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。