floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

♻️ 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
api/internal/storeLink/modelarts.go
View File

@ -19,13 +19,6 @@ type ModelArtsLink struct {
participant *models.StorelinkCenter 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 { 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} 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) { func (o *ModelArtsLink) DeleteImage(imageId string) (interface{}, error) {
// TODO modelArts删除镜像 // TODO modelArts删除镜像
return nil, nil 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) { func (o *ModelArtsLink) SubmitTask(imageId string, cmd string, envs []string, params []string, resourceId string) (interface{}, error) {
// modelArts提交任务 // modelArts提交任务
environments := make(map[string]string) environments := make(map[string]string)
parameters := make([]*modelarts.ParametersTrainJob, 0)
for _, param := range params { for _, env := range envs {
s := strings.Split(param, COMMA) s := strings.Split(env, COMMA)
environments[s[0]] = s[1] 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{ req := &modelarts.CreateTrainingJobReq{
Kind: "job", Kind: "job",
Metadata: &modelarts.MetadataS{ Metadata: &modelarts.MetadataS{
@ -83,6 +81,7 @@ func (o *ModelArtsLink) SubmitTask(imageId string, cmd string, envs []string, pa
}, },
Command: cmd, Command: cmd,
Environments: environments, Environments: environments,
Parameters: parameters,
}, },
Spec: &modelarts.SpecsC{ Spec: &modelarts.SpecsC{
Resource: &modelarts.ResourceCreateTraining{ Resource: &modelarts.ResourceCreateTraining{