Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.7.1
zouap 3 years ago
parent
commit
2f07dc3cdc
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      routers/repo/ai_model_convert.go

+ 6
- 6
routers/repo/ai_model_convert.go View File

@@ -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)


Loading…
Cancel
Save