创建任务运行
功能介绍
用于创建任务运行,获取运行id。
注意事项
调用本文API,需使用安全认证AK/SK鉴权,调用流程及鉴权介绍详见SDK安装及使用流程。
调用示例
from qianfan import resources
# 创建任务运行,具体参数请查看参数说明
create_job_resp = resources.FineTune.create_job(job={
"taskId": task_id,
"baseTrainType": "ERNIE-Bot-turbo",
"trainType": "ERNIE-Bot-turbo-0725",
"trainMode": "SFT",
"peftType": "LoRA",
"trainConfig": {
"epoch": 4,
"learningRate": 0.00002,
"batchSize": 4,
"maxSeqLen": 4096
},
"trainset": [
{
"type": 1,
"id": 1234
}
],
"trainsetRate": 20
})
# 获取运行 id
print(create_job_resp)
返回示例
QfResponse(code = 200, headers = {
...
},
body = {
'log_id': '3019477815',
'result': {
'id': 6301
}
},
)
参数说明
请求参数
名称 | 类型 | 必填 | 描述 |
---|---|---|---|
job | dict | 是 | 任务描述 |
job说明:
名称 | 类型 | 必填 | 描述 |
---|---|---|---|
taskId | int | 是 | 任务ID |
baseTrainType | string | 是 | 基础算法类型,可选值如下: · ERNIE-Bot-turbo · BLOOMZ-7B · Llama-2-7b |
trainType | string | 是 | 具体算法类型版本,说明: (1)如果baseTrainType为ERNIE-Bot-turbo,trainType可选值如下: · ERNIE-Bot-turbo-0516 · ERNIE-Bot-turbo-0704 · ERNIE-Bot-turbo-0725 (2)如果baseTrainType为BLOOMZ-7B,trainType为固定值BLOOMZ-7B (3)如果baseTrainType为Llama-2-7b,trainType为固定值Llama-2-7b |
trainMode | string | 是 | 训练方法,固定值:SFT |
peftType | string | 是 | parameter efficient finetuning方式,说明: (1)如果trainType为ERNIE-Bot-turbo-0516,peftType固定值为ALL (2)如果trainType为ERNIE-Bot-turbo-0725,peftType可选值如下: · ALL · LoRA (3)其他,peftType可选值如下: · ALL · P-tuning · LoRA |
trainConfig | dict | 是 | 训练配置 |
trainset | list | 是 | 数据集配置 |
trainsetRate | float | 是 | 数据拆分比例,取值范围:[0,50] |
description | string | 否 | 运行描述 |
trainConfig说明
名称 | 类型 | 必填 | 描述 |
---|---|---|---|
epoch | int | 是 | 迭代轮次,取值范围:[1, 50] |
learningRate | float | 是 | 学习率,说明: (1)如果trainType为ERNIE-Bot-turbo-0704,learningRate取值范围[0.00001,0.00004] (2)如果trainType为ERNIE-Bot-turbo-0725,learningRate取值范围[0.00001,0.00004] (3)其他,learningRate取值范围[0.0000002,0.0002] |
batchSize | int | 否 | 批处理大小,说明: (1)如果trainType为ERNIE-Bot-turbo-0516,batchSize取值范围:[16, 64] (2)如果trainType为ERNIE-Bot-turbo-0704或ERNIE-Bot-turbo-0725,该字段不需要填写 (3)其他,batchSize取值范围:[4, 64] |
maxSeqLen | int | 否 | 序列长度,说明: (1)可选值:4096 或 8192 (2)如果trainType为ERNIE-Bot-turbo-0725时,此字段必填 |
trainset说明
名称 | 类型 | 必填 | 描述 |
---|---|---|---|
type | int | 是 | 数据集类型,可选值如下: 1:表示平台数据集 2:表示bos数据集 |
id | int | 否 | 数据集版本ID,当type=1时,该字段必传 |
bosPath | string | 否 | 数据集版本bos路径,当type=2时,该字段必传 |
返回参数
名称 | 类型 | 描述 |
---|---|---|
log_id | int | 请求ID |
result | dict | 请求结果 |
result说明
名称 | 类型 | 描述 |
---|---|---|
id | int | 运行ID |
最后修改时间: 1 年前