From 728873b87e9076923fa4e0c48be975d4a91cd3ff Mon Sep 17 00:00:00 2001 From: zouap Date: Sat, 8 Oct 2022 11:05:01 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=EF=BC=8C?= =?UTF-8?q?=E8=A7=A3=E5=86=B3=E8=8E=B7=E5=8F=96=E5=8F=98=E9=87=8F=E9=97=AE?= =?UTF-8?q?=E9=A2=98=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/aisafety.go | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 00b87a63c..a26876895 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -665,17 +665,17 @@ func createForNPU(ctx *context.Context, jobName string) { } func createForGPU(ctx *context.Context, jobName string) { - BootFile := ctx.Query("BootFile") - displayJobName := ctx.Query("DisplayJobName") - description := ctx.Query("Description") - image := strings.TrimSpace(ctx.Query("Image")) - srcDataset := ctx.Query("srcDataset") //uuid - combatDataset := ctx.Query("combatDataset") //uuid + BootFile := ctx.Query("boot_file") + displayJobName := ctx.Query("display_job_name") + description := ctx.Query("description") + image := strings.TrimSpace(ctx.Query("image")) + srcDataset := ctx.Query("src_dataset") //uuid + combatDataset := ctx.Query("combat_dataset") //uuid evaluationIndex := ctx.Query("evaluationIndex") - Params := ctx.Query("RunParaList") - specId := ctx.QueryInt64("SpecId") - TrainUrl := ctx.Query("TrainUrl") - CkptName := ctx.Query("CkptName") + Params := ctx.Query("run_para_list") + specId := ctx.QueryInt64("spec_id") + TrainUrl := ctx.Query("train_url") + CkptName := ctx.Query("ckpt_name") ckptUrl := setting.Attachment.Minio.RealPath + TrainUrl + CkptName log.Info("ckpt url:" + ckptUrl) spec, err := resource.GetAndCheckSpec(ctx.User.ID, specId, models.FindSpecsOptions{ @@ -710,6 +710,7 @@ func createForGPU(ctx *context.Context, jobName string) { uuid := srcDataset + ";" + combatDataset datasetInfos, datasetNames, err := models.GetDatasetInfo(uuid) + log.Info("uuid=" + uuid) if err != nil { log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"]) modelSafetyNewDataPrepare(ctx)