diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go
index d133bf7a4..df04b7072 100644
--- a/routers/repo/ai_model_convert.go
+++ b/routers/repo/ai_model_convert.go
@@ -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,