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