diff --git a/modules/auth/modelarts.go b/modules/auth/modelarts.go index ced5ea1e8..9744bc387 100755 --- a/modules/auth/modelarts.go +++ b/modules/auth/modelarts.go @@ -33,6 +33,7 @@ type CreateModelArtsTrainJobForm struct { DisplayJobName string `form:"display_job_name" binding:"Required"` JobName string `form:"job_name" binding:"Required"` Attachment string `form:"attachment" binding:"Required"` + DatasetName string `form:"dataset_name"` BootFile string `form:"boot_file" binding:"Required"` WorkServerNumber int `form:"work_server_number" binding:"Required"` EngineID int `form:"engine_id" binding:"Required"` diff --git a/templates/repo/modelarts/trainjob/new.tmpl b/templates/repo/modelarts/trainjob/new.tmpl index df1fb90a9..8e2810317 100755 --- a/templates/repo/modelarts/trainjob/new.tmpl +++ b/templates/repo/modelarts/trainjob/new.tmpl @@ -196,8 +196,8 @@
- {{if .bootFile}} - + {{if .boot_file}} + {{else}} {{end}}