Browse Source

提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.21.12.1^2
zouap 3 years ago
parent
commit
31589e17a0
1 changed files with 8 additions and 5 deletions
  1. +8
    -5
      routers/repo/ai_model_manage.go

+ 8
- 5
routers/repo/ai_model_manage.go View File

@@ -7,7 +7,9 @@ import (
"io"
"io/ioutil"
"os"
"path"
"path/filepath"
"strings"

"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/context"
@@ -151,17 +153,18 @@ func downloadModelFromCloudBrainTwo(modelUUID string, jobName string, parentDir
models.AttachmentRelativePath(modelUUID) +
"/"

models, err := storage.GetObsListObject(jobName, parentDir)
modelDbResult, err := storage.GetObsListObject(jobName, parentDir)
if err != nil {
log.Info("get TrainJobListModel failed:", err)
return "", 0, err
}
if len(models) == 0 {
if len(modelDbResult) == 0 {
return "", 0, errors.New("cannot create model, as model is empty.")
}

for _, modelFile := range models {
log.Info("copy file, bucket=%s, src keyname=%s,dest keyname=%s", setting.Bucket, modelFile.ParenDir+modelFile.FileName, dataActualPath+modelFile.FileName)
prefix := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, jobName, setting.OutPutPath, parentDir), "/")
for _, modelFile := range modelDbResult {
destKeyName := "/aimodels/" + models.AttachmentRelativePath(modelUUID) + "/"
log.Info("copy file, bucket=%s, src keyname=%s, dest keyname=%s,", setting.Bucket, prefix+modelFile.FileName, destKeyName)
// err := storage.ObsCopyFile(setting.Bucket, modelFile.ParenDir+modelFile.FileName, setting.Bucket, dataActualPath+modelFile.FileName)
// if err != nil {
// log.Info("copy failed.")


Loading…
Cancel
Save