OpenAI ChatGPT模型训练

打开VSCode

新建一个工作目录

使用pip install --upgrade openai

配置环境变量:

OPENAI_API_KEY=<你的open ai key>

windows配置:(需要重启)

setx OPENAI_API_KEY "你的open ai key"

准备训练数据集文件:

格式如下:

放到工作目录下比如A.jsonl

使用命令格式化数据:

 

openai tools fine_tunes.prepare_data -f <LOCAL_FILE>

如:openai tools fine_tunes.prepare_data -f A.json.nl

然后一路Y即可 生成两个文件train和valid

然后创建训练模型:

openai api fine_tunes.create -t <TRAIN_FILE_ID_OR_PATH> -m <BASE_MODEL>

如:openai api fine_tunes.create -t ABC_prepared_train.jsonl -m curie

可以自定义模型名称--suffix:

openai api fine_tunes.create -t test.jsonl -m ada --suffix "custom model name"

正在训练...

等待训练完成(10-20分钟左右)训练完成如下:

消耗$0.04

模型名称:curie:ft-personal-2023-02-12-06-15-31

测试一下:

取训练数据集里的一条试试

测试结果如下:

测试通过~

本文链接:https://my.lmcjl.com/post/8021.html

展开阅读全文

4 评论

留下您的评论.