From 7206f81f0e34809b1267a77550de0a94aabcc15a Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 6 Jun 2022 16:53:47 +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 | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index 01004586b..1da736aa1 100644 --- a/routers/repo/ai_model_convert.go +++ b/routers/repo/ai_model_convert.go @@ -99,6 +99,7 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m mkModelPath(modelPath) uploadCodeToMinio(modelPath, modelConvert.ID, ModelMountPath+"/") command := getModelConvertCommand(modelConvert.ID, modelConvert.ModelPath) + log.Info("command=" + command) dataActualPath := setting.Attachment.Minio.RealPath + modelRelativePath if TrainResourceSpecs == nil { @@ -170,7 +171,7 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m func getModelConvertCommand(name string, modelFile string) string { var command string bootFile := "convert_pytorch.py" - command += "python /code/" + bootFile + " --model " + modelFile + " > " + ModelMountPath + "/" + name + "-" + LogFile + command += "python3 /code/" + bootFile + " --model " + modelFile + " > " + ModelMountPath + "/" + name + "-" + LogFile return command }