diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index 642c15cec..b9e911bad 100644 --- a/routers/repo/ai_model_convert.go +++ b/routers/repo/ai_model_convert.go @@ -24,7 +24,7 @@ const ( LogFile = "log.txt" DefaultBranchName = "master" SubTaskName = "task1" - GpuQueue = "openidebug" + GpuQueue = "openidgx" Success = "S000" 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+"/") 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 + "/" log.Info("modelPath=" + modelPath) mkModelPath(modelPath) + uploadCodeToMinio(modelPath, modelConvert.ID, ModelMountPath+"/") command := getModelConvertCommand(modelConvert.ID, modelConvert.ModelPath) log.Info("command=" + command) dataActualPath := setting.Attachment.Minio.RealPath + modelRelativePath + log.Info("dataActualPath=" + dataActualPath) if TrainResourceSpecs == nil { json.Unmarshal([]byte(setting.TrainResourceSpecs), &TrainResourceSpecs) @@ -137,7 +141,7 @@ func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, m Volumes: []models.Volume{ { HostPath: models.StHostPath{ - Path: codePath, + Path: minioCodePath, MountPath: CodeMountPath, ReadOnly: false, },