Browse Source

提交代码。

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

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

@@ -203,7 +203,7 @@ func createNpuTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context
engineId = int64(NPU_TENSORFLOW_IMAGE_ID)
bootfile = TensorFlowNpuBootFile
}
userCommand := "/bin/bash /home/work/run_train.sh " + codeObsPath + " /code/" + bootfile
userCommand := "/bin/bash /home/work/run_train.sh 's3://" + codeObsPath + "' '/code/" + bootfile + "' '/tmp/log/train.log' --'data_url'='s3://" + dataPath + "' --'train_url'='s3://" + outputObsPath + "'"
req := &modelarts.GenerateTrainJobReq{
JobName: modelConvert.ID,
DisplayJobName: modelConvert.Name,


Loading…
Cancel
Save