| @@ -24,7 +24,7 @@ const ( | |||||
| LogFile = "log.txt" | LogFile = "log.txt" | ||||
| DefaultBranchName = "master" | DefaultBranchName = "master" | ||||
| SubTaskName = "task1" | SubTaskName = "task1" | ||||
| GpuQueue = "openidgx" | |||||
| GpuQueue = "openidebug" | |||||
| 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" | ||||
| ) | ) | ||||
| @@ -91,7 +91,9 @@ func SaveModelConvert(ctx *context.Context) { | |||||
| func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error { | func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error { | ||||
| repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.ID) | repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.ID) | ||||
| if modelConvert.SrcEngine == PYTORCH_ENGINE { | if modelConvert.SrcEngine == PYTORCH_ENGINE { | ||||
| codePath := setting.JobPath + modelConvert.ID + CodeMountPath | codePath := setting.JobPath + modelConvert.ID + CodeMountPath | ||||
| downloadCode(repo, codePath, DefaultBranchName) | downloadCode(repo, codePath, DefaultBranchName) | ||||
| uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/") | uploadCodeToMinio(codePath+"/", modelConvert.ID, CodeMountPath+"/") | ||||