Browse Source

add default param

pull/552/head
tzwang 2 months ago
parent
commit
579d956f9a
3 changed files with 13 additions and 0 deletions
  1. +7
    -0
      internal/storeLink/modelarts.go
  2. +5
    -0
      internal/storeLink/openi.go
  3. +1
    -0
      internal/storeLink/storeLink.go

+ 7
- 0
internal/storeLink/modelarts.go View File

@@ -161,6 +161,13 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri
inputs := make([]*modelarts.InputTraining, 0)
outputs := make([]*modelarts.OutputTraining, 0)
outputValue := ""

// add default param, env
current_id := strconv.Itoa(int(m.participantId))
current_platform := CURRENT_PLATFORM + COMMA + current_id
params = append(params, current_platform)
envs = append(envs, current_platform)

for _, env := range envs {
// 找到第一个逗号位置
idx := strings.Index(env, COMMA)


+ 5
- 0
internal/storeLink/openi.go View File

@@ -216,6 +216,11 @@ func (o *OpenI) SubmitTask(ctx context.Context, imageId string, cmd string, envs
} `json:"parameter"`
}

// add default param
current_id := strconv.Itoa(int(o.participantId))
current_platform := CURRENT_PLATFORM + COMMA + current_id
params = append(params, current_platform)

for _, param := range params {
s := strings.Split(param, COMMA)
st := struct {


+ 1
- 0
internal/storeLink/storeLink.go View File

@@ -88,6 +88,7 @@ const (
POINT = "point"
RUNNINGTASK = "RUNNING_TASK"
RUNNING = "RUNNING"
CURRENT_PLATFORM = "CURRENT_PLATFORM" //default param
)

var (


Loading…
Cancel
Save