Browse Source

mod npu command

npu-train-new
lewis 2 years ago
parent
commit
86c69fabba
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      routers/repo/grampus.go

+ 2
- 2
routers/repo/grampus.go View File

@@ -1308,7 +1308,7 @@ func GrampusTrainJobShow(ctx *context.Context) {
taskList := make([]*models.Cloudbrain, 0)
taskList = append(taskList, task)
prepareSpec4Show(ctx, task)
ctx.Data["version_list_task"] = taskList
ctx.Data["datasetDownload"] = GetCloudBrainDataSetInfo(task.Uuid, task.DatasetName, false)
ctx.Data["canDownload"] = cloudbrain.CanModifyJob(ctx, task)
@@ -1469,7 +1469,7 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo
var commandCode string
if processorType == grampus.ProcessorTypeNPU {
paramCode += " --model_url=" + modelRemoteObsUrl
commandCode = "/bin/bash /home/work/run_train_for_openi.sh /home/work/openi.py " + grampus.NpuLocalLogUrl + paramCode + ";"
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


Loading…
Cancel
Save