From fc86884ed72e076695c50eed77eca54089a1cfce Mon Sep 17 00:00:00 2001 From: ychao_1983 Date: Mon, 19 Sep 2022 15:04:00 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- routers/repo/grampus.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) 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