| @@ -24,7 +24,7 @@ const ( | |||||
| LogFile = "log.txt" | LogFile = "log.txt" | ||||
| DefaultBranchName = "master" | DefaultBranchName = "master" | ||||
| SubTaskName = "task1" | SubTaskName = "task1" | ||||
| GpuQueue = "openidebug" | |||||
| GpuQueue = "openidgx" | |||||
| Success = "S000" | Success = "S000" | ||||
| GPU_PYTORCH_IMAGE = "dockerhub.pcl.ac.cn:5000/user-images/openi:tensorRT_7_zouap" | GPU_PYTORCH_IMAGE = "dockerhub.pcl.ac.cn:5000/user-images/openi:tensorRT_7_zouap" | ||||
| ) | ) | ||||
| @@ -100,14 +100,18 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m | |||||
| uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/") | uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/") | ||||
| log.Info("minio code path=" + setting.CBCodePathPrefix + modelConvert.ID) | log.Info("minio code path=" + setting.CBCodePathPrefix + modelConvert.ID) | ||||
| minioCodePath := setting.Attachment.Minio.RealPath + setting.Attachment.Minio.Bucket + "/" + setting.CBCodePathPrefix + modelConvert.ID | |||||
| modelPath := setting.JobPath + modelConvert.ID + ModelMountPath + "/" | modelPath := setting.JobPath + modelConvert.ID + ModelMountPath + "/" | ||||
| log.Info("modelPath=" + modelPath) | log.Info("modelPath=" + modelPath) | ||||
| mkModelPath(modelPath) | mkModelPath(modelPath) | ||||
| uploadCodeToMinio(modelPath, modelConvert.ID, ModelMountPath+"/") | uploadCodeToMinio(modelPath, modelConvert.ID, ModelMountPath+"/") | ||||
| command := getModelConvertCommand(modelConvert.ID, modelConvert.ModelPath) | command := getModelConvertCommand(modelConvert.ID, modelConvert.ModelPath) | ||||
| log.Info("command=" + command) | log.Info("command=" + command) | ||||
| dataActualPath := setting.Attachment.Minio.RealPath + modelRelativePath | dataActualPath := setting.Attachment.Minio.RealPath + modelRelativePath | ||||
| log.Info("dataActualPath=" + dataActualPath) | |||||
| if TrainResourceSpecs == nil { | if TrainResourceSpecs == nil { | ||||
| json.Unmarshal([]byte(setting.TrainResourceSpecs), &TrainResourceSpecs) | json.Unmarshal([]byte(setting.TrainResourceSpecs), &TrainResourceSpecs) | ||||
| @@ -137,7 +141,7 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m | |||||
| Volumes: []models.Volume{ | Volumes: []models.Volume{ | ||||
| { | { | ||||
| HostPath: models.StHostPath{ | HostPath: models.StHostPath{ | ||||
| Path: codePath, | |||||
| Path: minioCodePath, | |||||
| MountPath: CodeMountPath, | MountPath: CodeMountPath, | ||||
| ReadOnly: false, | ReadOnly: false, | ||||
| }, | }, | ||||