提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/repo/aisafety.go

@@ -526,6 +526,8 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) {
 func AiSafetyCreateForGetGrampusGPU(ctx *context.Context) {
 	ctx.Data["PageIsCloudBrain"] = true
 	ctx.Data["IsCreate"] = true
+	ctx.Data["type"] = models.TypeC2Net
+	ctx.Data["compute_resource"] = models.GPUResource
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 	ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName
 	ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID
@@ -542,7 +544,8 @@ func AiSafetyCreateForGetGrampusGPU(ctx *context.Context) {
 func AiSafetyCreateForGetGrampusNPU(ctx *context.Context) {
 	ctx.Data["PageIsCloudBrain"] = true
 	ctx.Data["IsCreate"] = true
-
+	ctx.Data["type"] = models.TypeC2Net
+	ctx.Data["compute_resource"] = models.NPUResource
 	ctx.Data["datasetType"] = models.TypeCloudBrainTwo
 	ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName
 	ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID
@@ -678,7 +681,7 @@ func createForGrampusGPU(ctx *context.Context, jobName string) {
 	image := strings.TrimSpace(ctx.Query("image"))
 	srcDataset := ctx.Query("src_dataset")       //uuid
 	combatDataset := ctx.Query("combat_dataset") //uuid
-	evaluationIndex := ctx.Query("evaluationIndex")
+	evaluationIndex := ctx.Query("evaluation_index")
 	Params := ctx.Query("run_para_list")
 	specId := ctx.QueryInt64("spec_id")
 	TrainUrl := ctx.Query("train_url")
@@ -828,7 +831,7 @@ func createForNPU(ctx *context.Context, jobName string) {
 
 	srcDataset := ctx.Query("src_dataset")       //uuid
 	combatDataset := ctx.Query("combat_dataset") //uuid
-	evaluationIndex := ctx.Query("evaluationIndex")
+	evaluationIndex := ctx.Query("evaluation_index")
 	Params := ctx.Query("run_para_list")
 	specId := ctx.QueryInt64("spec_id")
 
@@ -1018,7 +1021,7 @@ func createForGPU(ctx *context.Context, jobName string) {
 	image := strings.TrimSpace(ctx.Query("image"))
 	srcDataset := ctx.Query("src_dataset")       //uuid
 	combatDataset := ctx.Query("combat_dataset") //uuid
-	evaluationIndex := ctx.Query("evaluationIndex")
+	evaluationIndex := ctx.Query("evaluation_index")
 	Params := ctx.Query("run_para_list")
 	specId := ctx.QueryInt64("spec_id")
 	TrainUrl := ctx.Query("train_url")
@@ -1146,7 +1149,7 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error {
 	ctx.Data["image"] = strings.TrimSpace(ctx.Query("image"))
 	ctx.Data["src_dataset"] = ctx.Query("src_dataset")       //uuid
 	ctx.Data["combat_dataset"] = ctx.Query("combat_dataset") //uuid
-	ctx.Data["evaluationIndex"] = ctx.Query("evaluationIndex")
+	ctx.Data["evaluationIndex"] = ctx.Query("evaluation_index")
 	ctx.Data["run_para_list"] = ctx.Query("run_para_list")
 	ctx.Data["spec_id"] = ctx.QueryInt64("spec_id")
 	ctx.Data["train_url"] = ctx.Query("train_url")