From 47f03376222fe8495efdf081c509f726377cf1a5 Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 6 Jun 2022 17:36:16 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/ai_model_convert.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index 642c15cec..b9e911bad 100644 --- a/routers/repo/ai_model_convert.go +++ b/routers/repo/ai_model_convert.go @@ -24,7 +24,7 @@ const ( LogFile = "log.txt" DefaultBranchName = "master" SubTaskName = "task1" - GpuQueue = "openidebug" + GpuQueue = "openidgx" Success = "S000" GPU_PYTORCH_IMAGE = "dockerhub.pcl.ac.cn:5000/user-images/openi:tensorRT_7_zouap" ) @@ -100,14 +100,18 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/") log.Info("minio code path=" + setting.CBCodePathPrefix + modelConvert.ID) + minioCodePath := setting.Attachment.Minio.RealPath + setting.Attachment.Minio.Bucket + "/" + setting.CBCodePathPrefix + modelConvert.ID + modelPath := setting.JobPath + modelConvert.ID + ModelMountPath + "/" log.Info("modelPath=" + modelPath) mkModelPath(modelPath) + uploadCodeToMinio(modelPath, modelConvert.ID, ModelMountPath+"/") command := getModelConvertCommand(modelConvert.ID, modelConvert.ModelPath) log.Info("command=" + command) dataActualPath := setting.Attachment.Minio.RealPath + modelRelativePath + log.Info("dataActualPath=" + dataActualPath) if TrainResourceSpecs == nil { json.Unmarshal([]byte(setting.TrainResourceSpecs), &TrainResourceSpecs) @@ -137,7 +141,7 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m Volumes: []models.Volume{ { HostPath: models.StHostPath{ - Path: codePath, + Path: minioCodePath, MountPath: CodeMountPath, ReadOnly: false, },