Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.21.12.1^2
zouap 3 years ago
parent
commit
e36e756f5e
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      routers/repo/ai_model_manage.go

+ 3
- 1
routers/repo/ai_model_manage.go View File

@@ -18,7 +18,7 @@ import (
)

func SaveModelByParameters(trainTaskId int64, name string, version string, label string, description string, userId int64) {
aiTasks, _, err := models.Cloudbrains(&models.CloudbrainsOptions{
aiTasks, count, err := models.Cloudbrains(&models.CloudbrainsOptions{
JobID: trainTaskId,
Type: -1,
})
@@ -27,6 +27,8 @@ func SaveModelByParameters(trainTaskId int64, name string, version string, label
//ctx.Error(500, fmt.Sprintf("query cloud brain train task error. %v", err))
return
}
log.Info("query return count=" + fmt.Sprint(count))

uuid := uuid.NewV4()
id := uuid.String()
modelPath := id


Loading…
Cancel
Save