|
|
|
@@ -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+"/") |
|
|
|
|