diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go index 39601a600..8db1c11bf 100755 --- a/modules/modelarts/modelarts.go +++ b/modules/modelarts/modelarts.go @@ -66,6 +66,7 @@ type GenerateTrainJobReq struct { Description string CodeObsPath string BootFile string + BootFileUrl string DataUrl string TrainUrl string FlavorCode string @@ -89,6 +90,7 @@ type GenerateTrainJobVersionReq struct { Description string CodeObsPath string BootFile string + BootFileUrl string DataUrl string TrainUrl string FlavorCode string @@ -210,7 +212,7 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (err error Config: models.Config{ WorkServerNum: req.WorkServerNumber, AppUrl: req.CodeObsPath, - BootFileUrl: req.BootFile, + BootFileUrl: req.BootFileUrl, DataUrl: req.DataUrl, EngineID: req.EngineID, TrainUrl: req.TrainUrl, @@ -278,7 +280,7 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobVersionR Config: models.TrainJobVersionConfig{ WorkServerNum: req.WorkServerNumber, AppUrl: req.CodeObsPath, - BootFileUrl: req.BootFile, + BootFileUrl: req.BootFileUrl, DataUrl: req.DataUrl, EngineID: req.EngineID, TrainUrl: req.TrainUrl, diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index bf4a1f31f..5ca28515e 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -848,7 +848,8 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) DataUrl: dataPath, Description: description, CodeObsPath: codeObsPath, - BootFile: codeObsPath + bootFile, + BootFileUrl: codeObsPath + bootFile, + BootFile: bootFile, TrainUrl: outputObsPath, FlavorCode: flavorCode, WorkServerNumber: workServerNumber, @@ -912,9 +913,6 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) { ctx.Data["PageIsTrainJob"] = true var jobID = ctx.Params(":jobid") - // var fatherVersionName = ctx.Query("versionName") - // jobID = "19373" - // versionName = "V0009" jobName := form.JobName uuid := form.Attachment @@ -1072,7 +1070,8 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ DataUrl: dataPath, Description: description, CodeObsPath: codeObsPath, - BootFile: codeObsPath + bootFile, + BootFileUrl: codeObsPath + bootFile, + BootFile: bootFile, TrainUrl: outputObsPath, FlavorCode: flavorCode, WorkServerNumber: workServerNumber,