|
|
@@ -20,6 +20,7 @@ import ( |
|
|
|
func SaveModelByParameters(trainTaskId int64, name string, version string, label string, description string, userId int64) { |
|
|
|
aiTasks, _, err := models.Cloudbrains(&models.CloudbrainsOptions{ |
|
|
|
JobID: trainTaskId, |
|
|
|
Type: -1, |
|
|
|
}) |
|
|
|
if err != nil { |
|
|
|
log.Info("query task error." + err.Error()) |
|
|
|