提交代码。增加ModelArts自定义镜像实现。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/repo/modelarts.go

@@ -1190,6 +1190,9 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		TotalVersionCount: modelarts.TotalVersionCount,
 		DatasetName:       datasetNames,
 	}
+	userCommand, userImageUrl := getUserCommand(engineID, req)
+	req.UserCommand = userCommand
+	req.UserImageUrl = userImageUrl
 
 	//将params转换Parameters.Parameter，出错时返回给前端
 	var Parameters modelarts.Parameters
@@ -1208,7 +1211,7 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 	ctx.Redirect(setting.AppSubURL + ctx.Repo.RepoLink + "/modelarts/train-job")
 }
 
-func getUserCommand(engineId int64, req *modelarts.GenerateTrainJobReq) (string, string) {
+func getUserCommand(engineId int, req *modelarts.GenerateTrainJobReq) (string, string) {
 	userImageUrl := ""
 	userCommand := ""
 	if engineId < 0 {
@@ -1218,11 +1221,14 @@ func getUserCommand(engineId int64, req *modelarts.GenerateTrainJobReq) (string,
 			log.Info("json.Unmarshal failed:", err)
 		}
 		for _, engine := range engines.Info {
-			if engine.ID == int(engineId) {
+			if engine.ID == engineId {
 				userImageUrl = engine.Url
 				break
 			}
 		}
+		for _, param := range req.Parameters {
+			userCommand += " --'" + param.Label + "'='" + param.Value + "'"
+		}
 		return userCommand, userImageUrl
 	}
 	return userCommand, userImageUrl
@@ -1458,6 +1464,9 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 		TotalVersionCount: latestTask.TotalVersionCount + 1,
 		DatasetName:       datasetNames,
 	}
+	userCommand, userImageUrl := getUserCommand(engineID, req)
+	req.UserCommand = userCommand
+	req.UserImageUrl = userImageUrl
 
 	err = modelarts.GenerateTrainJobVersion(ctx, req, jobID)
 	if err != nil {