|
|
@@ -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 |
|
|
|