fix 2354

routers/repo/grampus.go

@@ -89,6 +89,12 @@ func grampusTrainJobNewDataPrepare(ctx *context.Context, processType string) err
 
 	ctx.Data["branchName"] = ctx.Repo.BranchName
 
+	if processType == grampus.ProcessorTypeGPU {
+		ctx.Data["datasetType"] = models.TypeCloudBrainOne
+	} else if processType == grampus.ProcessorTypeNPU {
+		ctx.Data["datasetType"] = models.TypeCloudBrainTwo
+	}
+
 	return nil
 }