diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go index 101b265bf..551fe78a6 100755 --- a/modules/modelarts/modelarts.go +++ b/modules/modelarts/modelarts.go @@ -393,11 +393,17 @@ func GenerateModelConvertTrainJob(req *GenerateTrainJobReq) (*models.CreateTrain Description: req.Description, Config: models.UserImageConfig{ WorkServerNum: req.WorkServerNumber, + AppUrl: req.CodeObsPath, + BootFileUrl: req.BootFileUrl, + DataUrl: req.DataUrl, + TrainUrl: req.TrainUrl, + LogUrl: req.LogUrl, PoolID: req.PoolID, CreateVersion: true, Flavor: models.Flavor{ Code: req.FlavorCode, }, + Parameter: req.Parameters, UserImageUrl: req.UserImageUrl, UserCommand: req.UserCommand, },