diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 9f985e269..5e8053ef2 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -324,6 +324,12 @@ func cloudBrainTrainJobErrorPrepare(ctx *context.Context, form auth.CreateCloudB
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 
@@ -439,6 +445,12 @@ func cloudBrainInferenceJobErrorPrepare(ctx *context.Context, form auth.CreateCl
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 
diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go
index ed37a542b..1c0d79952 100755
--- a/routers/repo/grampus.go
+++ b/routers/repo/grampus.go
@@ -204,6 +204,12 @@ func grampusTrainJobErrorPrepare(ctx *context.Context, processType string, form
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["image_id"] = form.ImageID