Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.7.1
zouap 3 years ago
parent
commit
133e080baa
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      routers/repo/ai_model_convert.go

+ 3
- 1
routers/repo/ai_model_convert.go View File

@@ -24,7 +24,7 @@ const (
LogFile = "log.txt"
DefaultBranchName = "master"
SubTaskName = "task1"
GpuQueue = "openidgx"
GpuQueue = "openidebug"
Success = "S000"
GPU_PYTORCH_IMAGE = "dockerhub.pcl.ac.cn:5000/user-images/openi:tensorRT_7_zouap"
)
@@ -91,7 +91,9 @@ func SaveModelConvert(ctx *context.Context) {
func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error {
repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.ID)
if modelConvert.SrcEngine == PYTORCH_ENGINE {

codePath := setting.JobPath + modelConvert.ID + CodeMountPath

downloadCode(repo, codePath, DefaultBranchName)

uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/")


Loading…
Cancel
Save