diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go index 76f2bd98c..c48950915 100755 --- a/routers/repo/grampus.go +++ b/routers/repo/grampus.go @@ -787,8 +787,17 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo } } - commandCode := "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" - command += commandCode + if processorType == grampus.ProcessorTypeNPU { + commandCode := "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" + commandCode = "/bin/bash /home/work/run_train_for_openi.sh " + workDir + "code/" + strings.ToLower(repoName) + "/" + bootFile + " /tmp/log/train.log" + paramCode + ";" + command += commandCode + } else if processorType == grampus.ProcessorTypeGPU { + commandCode := "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" + command += commandCode + } + + //commandCode := "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" + //command += commandCode //get exec result commandGetRes := "result=$?;"