|
|
@@ -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, |
|
|
|
}, |
|
|
|