提交代码

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