♻️ modelarts 提交任务参数修改

Signed-off-by: devad <cossjie@foxmail.com>

Former-commit-id: bfa8281d3bceffb53a87900a276c7679f6c800cd
This commit is contained in:
devad 2023-11-02 10:37:51 +08:00
parent 795c6d934e
commit 1e89049f10
1 changed files with 11 additions and 12 deletions

View File

@ -19,13 +19,6 @@ type ModelArtsLink struct {
participant *models.StorelinkCenter
}
//const (
// IMG_NAME_PREFIX = "oct_"
// IMG_VERSION_PREFIX = "version_"
// TASK_NAME_PREFIX = "trainJob_"
// RESOURCE_POOL = "common-pool"
//)
func NewModelArtsLink(ctx context.Context, svcCtx *svc.ServiceContext, participant *models.StorelinkCenter) *ModelArtsLink {
return &ModelArtsLink{ctx: ctx, svcCtx: svcCtx, participant: participant, platform: participant.Name, pageIndex: 1, pageSize: 100}
}
@ -37,7 +30,6 @@ func (o *ModelArtsLink) UploadImage(path string) (interface{}, error) {
func (o *ModelArtsLink) DeleteImage(imageId string) (interface{}, error) {
// TODO modelArts删除镜像
return nil, nil
}
@ -65,12 +57,18 @@ func (o *ModelArtsLink) QueryImageList() (interface{}, error) {
func (o *ModelArtsLink) SubmitTask(imageId string, cmd string, envs []string, params []string, resourceId string) (interface{}, error) {
// modelArts提交任务
environments := make(map[string]string)
for _, param := range params {
s := strings.Split(param, COMMA)
parameters := make([]*modelarts.ParametersTrainJob, 0)
for _, env := range envs {
s := strings.Split(env, COMMA)
environments[s[0]] = s[1]
}
for _, param := range params {
s := strings.Split(param, COMMA)
parameters = append(parameters, &modelarts.ParametersTrainJob{
Name: s[0],
Value: s[1],
})
}
req := &modelarts.CreateTrainingJobReq{
Kind: "job",
Metadata: &modelarts.MetadataS{
@ -83,6 +81,7 @@ func (o *ModelArtsLink) SubmitTask(imageId string, cmd string, envs []string, pa
},
Command: cmd,
Environments: environments,
Parameters: parameters,
},
Spec: &modelarts.SpecsC{
Resource: &modelarts.ResourceCreateTraining{