|
|
|
@@ -995,14 +995,13 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
if pretrainModelFileName != "" { |
|
|
|
paramCode += " --ckpt_url" + "=" + workDir + "pretrainmodel/" + pretrainModelFileName |
|
|
|
} |
|
|
|
|
|
|
|
var commandCode string |
|
|
|
if processorType == grampus.ProcessorTypeNPU { |
|
|
|
commandCode = "/bin/bash /home/work/run_train_for_openi.sh " + workDir + "code/" + strings.ToLower(repoName) + "/" + bootFile + " /tmp/log/train.log" + paramCode + ";" |
|
|
|
} else if processorType == grampus.ProcessorTypeGPU { |
|
|
|
if pretrainModelFileName != "" { |
|
|
|
paramCode += " --ckpt_url" + "=" + workDir + "pretrainmodel/" + pretrainModelFileName |
|
|
|
} |
|
|
|
commandCode = "cd " + workDir + "code/" + strings.ToLower(repoName) + ";python " + bootFile + paramCode + ";" |
|
|
|
} |
|
|
|
|
|
|
|
|