liuzx
3 years ago
1 changed files with 4 additions and 2 deletions
Unified View
Diff Options
+ 4
- 2
routers/repo/modelarts.go
View File
| @@ -54,8 +54,10 @@ func DebugJobIndex(ctx *context.Context) { | |||||
| page = 1 | page = 1 | ||||
| } | } | ||||
| debugType := modelarts.DebugType | debugType := modelarts.DebugType | ||||
| jobTypeNot := false | |||||
| if debugListType == models.GPUResource { | if debugListType == models.GPUResource { | ||||
| debugType = models.TypeCloudBrainOne | debugType = models.TypeCloudBrainOne | ||||
| jobTypeNot = true | |||||
| } else if debugListType == models.NPUResource { | } else if debugListType == models.NPUResource { | ||||
| debugType = models.TypeCloudBrainTwo | debugType = models.TypeCloudBrainTwo | ||||
| } | } | ||||
| @@ -67,8 +69,8 @@ func DebugJobIndex(ctx *context.Context) { | |||||
| }, | }, | ||||
| RepoID: repo.ID, | RepoID: repo.ID, | ||||
| Type: debugType, | Type: debugType, | ||||
| JobTypeNot: true, | |||||
| JobType: string(models.JobTypeTrain), | |||||
| JobTypeNot: jobTypeNot, | |||||
| JobType: string(models.JobTypeDebug), | |||||
| }) | }) | ||||
| if err != nil { | if err != nil { | ||||
| ctx.ServerError("Get debugjob faild:", err) | ctx.ServerError("Get debugjob faild:", err) | ||||