|
|
|
@@ -31,6 +31,7 @@ import ( |
|
|
|
const ( |
|
|
|
tplModelSafetyTestCreateGpu = "repo/modelsafety/newgpu" |
|
|
|
tplModelSafetyTestCreateNpu = "repo/modelsafety/newnpu" |
|
|
|
tplModelSafetyTestShow = "repo/modelsafety/show" |
|
|
|
) |
|
|
|
|
|
|
|
func CloudBrainAiSafetyCreateTest(ctx *context.Context) { |
|
|
|
@@ -67,13 +68,16 @@ func CloudBrainAiSafetyCreateTest(ctx *context.Context) { |
|
|
|
} |
|
|
|
|
|
|
|
func GetAiSafetyTaskByJob(job *models.Cloudbrain) { |
|
|
|
|
|
|
|
if job == nil { |
|
|
|
log.Error("GetCloudbrainByJobID failed") |
|
|
|
return |
|
|
|
} |
|
|
|
syncAiSafetyTaskStatus(job) |
|
|
|
} |
|
|
|
|
|
|
|
func GetAiSafetyTaskTmpl(ctx *context.Context) { |
|
|
|
ctx.Data["id"] = ctx.Params(":jobid") |
|
|
|
ctx.HTML(200, tplModelSafetyTestShow) |
|
|
|
} |
|
|
|
|
|
|
|
func GetAiSafetyTask(ctx *context.Context) { |
|
|
|
@@ -85,6 +89,8 @@ func GetAiSafetyTask(ctx *context.Context) { |
|
|
|
} |
|
|
|
syncAiSafetyTaskStatus(job) |
|
|
|
job, err = models.GetCloudbrainByJobIDWithDeleted(ID) |
|
|
|
job.BenchmarkType = "CV" |
|
|
|
job.BenchmarkTypeName = "Classification" |
|
|
|
ctx.JSON(200, job) |
|
|
|
} |
|
|
|
|
|
|
|
|