|
|
@@ -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 |
|
|
|