From 5fd169ab3b2c05a9416c20c1b84318e922009bde Mon Sep 17 00:00:00 2001 From: ychao_1983 Date: Wed, 22 Mar 2023 16:29:10 +0800 Subject: [PATCH] fix-3592 --- routers/repo/cloudbrain.go | 4 ++-- routers/repo/grampus.go | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 2374232009..af897107fa 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -2886,7 +2886,7 @@ func getInferenceJobCommand(form auth.CreateCloudBrainInferencForm) (string, err } } - param += " --modelname" + "=" + form.CkptName + param += " --modelname" + "='" + form.CkptName + "'" command += "python /code/" + bootFile + param + " > " + cloudbrain.ResultPath + "/" + form.DisplayJobName + "-" + cloudbrain.LogFile @@ -2917,7 +2917,7 @@ func getTrainJobCommand(form auth.CreateCloudBrainForm) (string, error) { } } if form.CkptName != "" { - param += " --ckpt_url" + "=" + "/pretrainmodel/" + form.CkptName + param += " --ckpt_url" + "=" + "'/pretrainmodel/" + form.CkptName + "'" } command += "python /code/" + bootFile + param + " > " + cloudbrain.ModelMountPath + "/" + form.DisplayJobName + "-" + cloudbrain.LogFile diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go index 93526fd986..8579ad1d9c 100755 --- a/routers/repo/grampus.go +++ b/routers/repo/grampus.go @@ -1754,12 +1754,12 @@ func generateCommand(repoName, processorType, bootFile, paramSrc, outputRemotePa commandCode = "source /home/ma-user/.bashrc;python /home/ma-user/davinci/train/davincirun.py python /home/ma-user/openi.py " + paramCode + ";" } else if processorType == grampus.ProcessorTypeGPU { if pretrainModelFileName != "" { - paramCode += " --ckpt_url" + "=" + workDir + "pretrainmodel/" + pretrainModelFileName + paramCode += " --ckpt_url" + "='" + workDir + "pretrainmodel/" + pretrainModelFileName+"'" } commandCode = "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" } else if processorType == grampus.ProcessorTypeGCU { if pretrainModelFileName != "" { - paramCode += " --ckpt_url" + "=" + workDir + "pretrainmodel/" + pretrainModelFileName + paramCode += " --ckpt_url" + "='" + workDir + "pretrainmodel/" + pretrainModelFileName+"'" } commandCode = "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python3 " + bootFile + paramCode + ";" } -- 2.34.1