提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/private/tool.go

@@ -46,7 +46,7 @@ func RepoStatisticManually(ctx *macaron.Context) {
 }
 
 func CreateModel(ctx *macaron.Context) {
-	trainTaskId := ctx.QueryInt64("TrainTask")
+	trainTaskId := ctx.Query("TrainTask")
 	name := ctx.Query("Name")
 	version := ctx.Query("Version")
 	label := ctx.Query("Label")

routers/repo/ai_model_manage.go

@@ -17,17 +17,13 @@ import (
 	uuid "github.com/satori/go.uuid"
 )
 
-func SaveModelByParameters(trainTaskId int64, name string, version string, label string, description string, userId int64) {
-	aiTasks, count, err := models.Cloudbrains(&models.CloudbrainsOptions{
-		JobID: trainTaskId,
-		Type:  -1,
-	})
+func SaveModelByParameters(trainTaskId string, name string, version string, label string, description string, userId int64) {
+	aiTask, err := models.GetCloudbrainByJobID(trainTaskId)
 	if err != nil {
 		log.Info("query task error." + err.Error())
 		//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()
@@ -36,12 +32,6 @@ func SaveModelByParameters(trainTaskId int64, name string, version string, label
 	var modelSize int64
 	cloudType := models.TypeCloudBrainTwo
 
-	if len(aiTasks) != 1 {
-		log.Info("query task error. len=" + fmt.Sprint(len(aiTasks)))
-		//ctx.Error(500, fmt.Sprintf("query cloud brain train task error. %v", err))
-		return
-	}
-	aiTask := aiTasks[0]
 	log.Info("find task name:" + aiTask.JobName)
 	aimodels := models.QueryModelByName(name, userId)
 	if len(aimodels) > 0 {
@@ -51,7 +41,7 @@ func SaveModelByParameters(trainTaskId int64, name string, version string, label
 			}
 		}
 	}
-	cloudType = aiTask.Cloudbrain.Type
+	cloudType = aiTask.Type
 	//download model zip   //train type
 	if cloudType == models.TypeCloudBrainTrainJob {
 		modelPath, modelSize, err = downloadModelFromCloudBrainTwo(id, aiTask.JobName, "")