diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index 6b184f098..1210bfb7e 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -301,6 +301,13 @@ type CreateNotebookParams struct {
 	Flavor      string     		`json:"flavor"`
 	Spec  		Spec 			`json:"spec"`
 	Workspace   Workspace   	`json:"workspace"`
+	Pool		Pool			`json:"pool"`
+}
+
+type Pool struct {
+	ID    	string     		`json:"id"`
+	Name 	string       	`json:"name"`
+	Type	string     		`json:"type"`
 }
 
 type Workspace struct {
diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go
index 8f4ad945e..34fd1195c 100755
--- a/modules/modelarts/modelarts.go
+++ b/modules/modelarts/modelarts.go
@@ -15,7 +15,11 @@ const (
 	storageTypeOBS = "obs"
 	autoStopDuration = 4 * 60 * 60
 	flavor = "modelarts.kat1.xlarge"
-	profileID = "Python3-ascend910-arm"
+	//profileID = "Python3-ascend910-arm"
+	profileID = "efa847c0-7359-11eb-b34f-0255ac100057"
+	poolID = "pool1328035d"
+	poolName = "train-private-1"
+	poolType = "USER_DEFINED"
 
 	DataSetMountPath = "/home/ma-user/work"
 	NotebookEnv = "Python3"
@@ -91,6 +95,11 @@ func GenerateTask(ctx *context.Context, jobName, uuid, description string) error
 		Description:description,
 		ProfileID:  profileID,
 		Flavor: flavor,
+		Pool: models.Pool{
+			ID: poolID,
+			Name: poolName,
+			Type: poolType,
+		},
 		Spec: models.Spec{
 			Storage: models.Storage{
 				Type: storageTypeOBS,