diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go index fc4cca516..023ba9a6a 100755 --- a/routers/repo/grampus.go +++ b/routers/repo/grampus.go @@ -146,13 +146,15 @@ func grampusTrainJobNewDataPrepare(ctx *context.Context, processType string) err ctx.Data["branch_name"] = ctx.Cloudbrain.BranchName ctx.Data["engine_name"] = ctx.Cloudbrain.EngineName ctx.Data["WorkServerNumber"] = ctx.Cloudbrain.WorkServerNumber - ctx.Data["image"] = ctx.Cloudbrain.Image + if ctx.Cloudbrain.Image != "" { + ctx.Data["image"] = ctx.Cloudbrain.Image + } else { + ctx.Data["image"] = ctx.Cloudbrain.EngineName + } ctx.Data["dataset_name"] = ctx.Cloudbrain.DatasetName ctx.Data["model_name"] = ctx.Cloudbrain.ModelName ctx.Data["model_version"] = ctx.Cloudbrain.ModelVersion - ctx.Data["dataset_name"] = ctx.Cloudbrain.DatasetName - ctx.Data["model_name"] = ctx.Cloudbrain.ModelName ctx.Data["ckpt_name"] = ctx.Cloudbrain.CkptName ctx.Data["label_names"] = ctx.Cloudbrain.LabelName ctx.Data["PreTrainModelUrl"] = ctx.Cloudbrain.PreTrainModelUrl diff --git a/templates/repo/cloudbrain/trainjob/versionnew.tmpl b/templates/repo/cloudbrain/trainjob/versionnew.tmpl index a622db8fb..6a625c35b 100644 --- a/templates/repo/cloudbrain/trainjob/versionnew.tmpl +++ b/templates/repo/cloudbrain/trainjob/versionnew.tmpl @@ -70,7 +70,7 @@
{{template "repo/header" .}}
- + {{template "base/alert" .}}

{{.i18n.Tr "repo.modelarts.train_job.new"}}