Browse Source

调整GPU训练任务的日志显示逻辑。#2706

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.8.2^2
zouap 3 years ago
parent
commit
434d29b808
1 changed files with 0 additions and 49 deletions
  1. +0
    -49
      routers/repo/cloudbrain.go

+ 0
- 49
routers/repo/cloudbrain.go View File

@@ -2885,52 +2885,3 @@ func getFlavorNameByFlavorCode(flavorCode string) string {
"核" + strconv.Itoa(cardNum*256) + "GB"
return cloudbrainTwoFlavorName
}

func GetLogFromModelDir(ctx *context.Context) {
prefix := "/" + setting.CBCodePathPrefix + ctx.Cloudbrain.JobName + ModelMountPath
files, err := storage.GetOneLevelAllObjectUnderDirMinio(setting.Attachment.Minio.Bucket, prefix, "")
if err != nil {
log.Error("query cloudbrain model failed: %v", err)
return
}
startLine := ctx.QueryInt("startLine")
endLine := ctx.QueryInt("endLine")

re := ""
for _, file := range files {
if strings.HasSuffix(file.FileName, "log.txt") {
path := storage.GetMinioPath(ctx.Cloudbrain.JobName+ModelMountPath+"/", file.FileName)
log.Info("path=" + path)
reader, err := os.Open(path)
defer reader.Close()
if err == nil {
r := bufio.NewReader(reader)
for i := 0; i < endLine; i++ {
line, error := r.ReadString('\n')
log.Info("line=" + line)
if error == io.EOF {
log.Info("read file completed.")
break
}
if error != nil {
log.Info("read file error." + error.Error())
break
}
if error == nil {
if i >= startLine {
re = re + line
}
}
}
} else {
log.Info("error:" + err.Error())
}
break
}
}

ctx.JSON(http.StatusOK, map[string]interface{}{
"JobName": ctx.Cloudbrain.JobName,
"Content": re,
})
}

Loading…
Cancel
Save