OpenAI ChatGPT模型训练

打开VSCode

新建一个工作目录

OpenAI ChatGPT模型训练_第1张图片

使用pip install --upgrade openai

配置环境变量:

OPENAI_API_KEY=<你的open ai key>

windows配置:(需要重启)

setx OPENAI_API_KEY "你的open ai key"

准备训练数据集文件:

格式如下:

OpenAI ChatGPT模型训练_第2张图片

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

使用命令格式化数据:

 OpenAI ChatGPT模型训练_第3张图片

openai tools fine_tunes.prepare_data -f

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

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

然后创建训练模型:

OpenAI ChatGPT模型训练_第4张图片

openai api fine_tunes.create -t -m

如: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"

正在训练...

e211e073bc8e6a27f1ea5e1c76b657f7.png

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

消耗$0.04

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

OpenAI ChatGPT模型训练_第5张图片

测试一下:

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

OpenAI ChatGPT模型训练_第6张图片

测试结果如下:

OpenAI ChatGPT模型训练_第7张图片

测试通过~

你可能感兴趣的:(AI,openai,GPT-3,模型训练,chatGPT模型微调,chatGPT)