|
|
|
@@ -115,7 +115,7 @@ func createNpuTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context |
|
|
|
codeObsPath := "/" + setting.Bucket + modelarts.JobPath + modelConvert.ID + modelarts.CodePath |
|
|
|
outputObsPath := "/" + setting.Bucket + modelarts.JobPath + modelConvert.ID + modelarts.OutputPath + VersionOutputPath + "/" |
|
|
|
logObsPath := "/" + setting.Bucket + modelarts.JobPath + modelConvert.ID + modelarts.LogPath + VersionOutputPath + "/" |
|
|
|
dataPath := modelRelativePath |
|
|
|
dataPath := "/" + modelRelativePath |
|
|
|
|
|
|
|
_, err := ioutil.ReadDir(codeLocalPath) |
|
|
|
if err == nil { |
|
|
|
@@ -150,28 +150,28 @@ func createNpuTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context |
|
|
|
} |
|
|
|
param := make([]models.Parameter, 0) |
|
|
|
modelPara := models.Parameter{ |
|
|
|
Label: "--model", |
|
|
|
Label: "model", |
|
|
|
Value: modelConvert.ModelPath, |
|
|
|
} |
|
|
|
param = append(param, modelPara) |
|
|
|
|
|
|
|
batchSizePara := models.Parameter{ |
|
|
|
Label: "--n", |
|
|
|
Label: "n", |
|
|
|
Value: fmt.Sprint(n), |
|
|
|
} |
|
|
|
param = append(param, batchSizePara) |
|
|
|
channelSizePara := models.Parameter{ |
|
|
|
Label: "--c", |
|
|
|
Label: "c", |
|
|
|
Value: fmt.Sprint(c), |
|
|
|
} |
|
|
|
param = append(param, channelSizePara) |
|
|
|
heightPara := models.Parameter{ |
|
|
|
Label: "--h", |
|
|
|
Label: "h", |
|
|
|
Value: fmt.Sprint(h), |
|
|
|
} |
|
|
|
param = append(param, heightPara) |
|
|
|
widthPara := models.Parameter{ |
|
|
|
Label: "--w", |
|
|
|
Label: "w", |
|
|
|
Value: fmt.Sprint(w), |
|
|
|
} |
|
|
|
param = append(param, widthPara) |
|
|
|
|