Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
pull/2480/head
zouap 3 years ago
parent
commit
47f0337622
1 changed files with 6 additions and 2 deletions
  1. +6
    -2
      routers/repo/ai_model_convert.go

+ 6
- 2
routers/repo/ai_model_convert.go View File

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


Loading…
Cancel
Save