Browse Source

add default param

tags/v1.21.12.1
lewis 4 years ago
parent
commit
2d09138509
2 changed files with 15 additions and 4 deletions
  1. +2
    -0
      modules/modelarts/modelarts.go
  2. +13
    -4
      routers/repo/modelarts.go

+ 2
- 0
modules/modelarts/modelarts.go View File

@@ -45,6 +45,8 @@ const (
JobPath = "/job/"
OrderDesc = "desc"
OrderAsc = "asc"
TrainUrl = "train_url"
DataUrl = "data_url"
)

type GenerateTrainJobReq struct {


+ 13
- 4
routers/repo/modelarts.go View File

@@ -388,6 +388,13 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)

var parameters models.Parameters
param := make([]models.Parameter, 0)
param = append(param, models.Parameter{
Label: modelarts.TrainUrl,
Value: outputObsPath,
}, models.Parameter{
Label: modelarts.DataUrl,
Value: dataPath,
})
if len(params) != 0 {
err := json.Unmarshal([]byte(params), &parameters)
if err != nil {
@@ -397,10 +404,12 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
}

for _, parameter := range parameters.Parameter {
param = append(param, models.Parameter{
Label: parameter.Label,
Value: parameter.Value,
})
if parameter.Label != modelarts.TrainUrl && parameter.Label != modelarts.DataUrl {
param = append(param, models.Parameter{
Label: parameter.Label,
Value: parameter.Value,
})
}
}
}



Loading…
Cancel
Save