Browse Source

fix bug

tags/v1.22.11.1^2
lewis 3 years ago
parent
commit
6478c32fbd
2 changed files with 13 additions and 9 deletions
  1. +11
    -7
      routers/api/v1/repo/modelarts.go
  2. +2
    -2
      routers/repo/grampus.go

+ 11
- 7
routers/api/v1/repo/modelarts.go View File

@@ -483,14 +483,18 @@ func ModelList(ctx *context.APIContext) {
status = models.StorageScheduleSucceed
} else {
if models.IsTrainJobTerminal(task.Status) {
record, _ := models.GetScheduleRecordByCloudbrainID(task.ID)
if record != nil {
status = record.Status
if status == models.StorageScheduleSucceed {
status = models.StorageNoFile
}
if task.Status == models.GrampusStatusStopped {
status = models.StorageNoFile
} else {
status = models.StorageScheduleProcessing
record, _ := models.GetScheduleRecordByCloudbrainID(task.ID)
if record != nil {
status = record.Status
if status == models.StorageScheduleSucceed {
status = models.StorageNoFile
}
} else {
status = models.StorageScheduleProcessing
}
}
} else {
status = models.StorageScheduleWaiting


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

@@ -922,7 +922,7 @@ func GrampusTrainJobShow(ctx *context.Context) {
ctx.Data["canDownload"] = cloudbrain.CanModifyJob(ctx, task)
ctx.Data["displayJobName"] = task.DisplayJobName

ctx.Data["ai_center"] = cloudbrainService.GetAiCenterShow(task.AiCenter,ctx)
ctx.Data["ai_center"] = cloudbrainService.GetAiCenterShow(task.AiCenter, ctx)

ctx.HTML(http.StatusOK, tplGrampusTrainJobShow)
}
@@ -1004,7 +1004,7 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo
//no need to process
} else if processorType == grampus.ProcessorTypeGPU {
unZipDatasetCommand := generateDatasetUnzipCommand(datasetName)
commandUnzip := "cd " + workDir + "code;unzip -q master.zip;echo \"start to unzip dataset\";cd " + workDir + "dataset;" + unZipDatasetCommand
commandUnzip := "cd " + workDir + "code;unzip -q master.zip;rm -f master.zip;echo \"start to unzip dataset\";cd " + workDir + "dataset;" + unZipDatasetCommand
command += commandUnzip
}



Loading…
Cancel
Save