diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index b4996a8ef..67885777c 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -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