From 133e080baa3347aff77a6dc3eeeedacd8c9a8ec5 Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 6 Jun 2022 17:12:59 +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 | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index 424e39484..7b5f7c6eb 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 = "openidgx" + GpuQueue = "openidebug" Success = "S000" GPU_PYTORCH_IMAGE = "dockerhub.pcl.ac.cn:5000/user-images/openi:tensorRT_7_zouap" ) @@ -91,7 +91,9 @@ func SaveModelConvert(ctx *context.Context) { func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error { repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.ID) if modelConvert.SrcEngine == PYTORCH_ENGINE { + codePath := setting.JobPath + modelConvert.ID + CodeMountPath + downloadCode(repo, codePath, DefaultBranchName) uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/")