diff --git a/routers/api/v1/repo/modelarts.go b/routers/api/v1/repo/modelarts.go index ab4fc5448..269b6431c 100755 --- a/routers/api/v1/repo/modelarts.go +++ b/routers/api/v1/repo/modelarts.go @@ -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 diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go index 2fb6d8318..049bb18c8 100755 --- a/routers/repo/grampus.go +++ b/routers/repo/grampus.go @@ -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 }