添加训练任务磁盘挂载

models/cloudbrain.go

@@ -1045,6 +1045,7 @@ type CreateUserImageTrainJobParams struct {
 	Description string          `json:"job_desc"`
 	Config      UserImageConfig `json:"config"`
 	WorkspaceID string          `json:"workspace_id"`
+	Volume      Volumes         `json:"volumes"`
 }
 
 type UserImageConfig struct {
@@ -1063,10 +1064,11 @@ type UserImageConfig struct {
 }
 
 type CreateTrainJobParams struct {
-	JobName     string `json:"job_name"`
-	Description string `json:"job_desc"`
-	Config      Config `json:"config"`
-	WorkspaceID string `json:"workspace_id"`
+	JobName     string  `json:"job_name"`
+	Description string  `json:"job_desc"`
+	Config      Config  `json:"config"`
+	WorkspaceID string  `json:"workspace_id"`
+	Volume      Volumes `json:"volumes"`
 }
 
 type Config struct {

modules/modelarts/modelarts.go

@@ -66,6 +66,9 @@ const (
 	SortByCreateTime  = "create_time"
 	ConfigTypeCustom  = "custom"
 	TotalVersionCount = 1
+
+	VolumeSourcePath = "192.168.0.30:/"
+	VolumeDestPath   = "/home/ma-user/sfs"
 )
 
 var (
@@ -374,6 +377,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 				UserImageUrl: req.UserImageUrl,
 				UserCommand:  req.UserCommand,
 			},
+			Volume: models.Volumes{
+				Nfs: models.Nfs{
+					SourcePath: VolumeSourcePath,
+					DestPath:   VolumeDestPath,
+					ReadOnly:   false,
+				},
+			},
 		})
 	} else {
 		jobResult, createErr = createTrainJob(models.CreateTrainJobParams{
@@ -394,6 +404,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 				},
 				Parameter: req.Parameters,
 			},
+			Volume: models.Volumes{
+				Nfs: models.Nfs{
+					SourcePath: VolumeSourcePath,
+					DestPath:   VolumeDestPath,
+					ReadOnly:   false,
+				},
+			},
 		})
 	}
 	if createErr != nil {