♻️ modelarts 提交任务参数修改
Signed-off-by: devad <cossjie@foxmail.com> Former-commit-id: bfa8281d3bceffb53a87900a276c7679f6c800cd
This commit is contained in:
parent
795c6d934e
commit
1e89049f10
|
@ -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{
|
||||
|
|
Loading…
Reference in New Issue