Browse Source

提交代码

tags/v1.22.12.1^2
ychao_1983 2 years ago
parent
commit
c1f719126b
2 changed files with 7 additions and 7 deletions
  1. +2
    -2
      routers/repo/grampus.go
  2. +5
    -5
      routers/repo/modelarts.go

+ 2
- 2
routers/repo/grampus.go View File

@@ -1171,8 +1171,8 @@ func GrampusNotebookShow(ctx *context.Context) {
ctx.HTML(http.StatusOK, tplGrampusNotebookShow)
}

func getDatasetDownloadInfo(ctx *context.Context, task *models.Cloudbrain) []models.DatasetDownload {
datasetDownload := make([]models.DatasetDownload, 0)
func getDatasetDownloadInfo(ctx *context.Context, task *models.Cloudbrain) []*models.DatasetDownload {
datasetDownload := make([]*models.DatasetDownload, 0)
if ctx.IsSigned {
if task.Uuid != "" && task.UserID == ctx.User.ID {
if task.IsGPUTask() {


+ 5
- 5
routers/repo/modelarts.go View File

@@ -306,7 +306,7 @@ func NotebookShow(ctx *context.Context) {

}

datasetDownload := make([]models.DatasetDownload, 0)
datasetDownload := make([]*models.DatasetDownload, 0)
var modelDownload models.ModelDownload
if ctx.IsSigned {
if task.Uuid != "" && task.UserID == ctx.User.ID {
@@ -361,8 +361,8 @@ func GetModelDownload(task *models.Cloudbrain) models.ModelDownload {
return modelDownload
}

func GetCloudBrainDataSetInfo(uuid string, datasetname string, isNeedDown bool) []models.DatasetDownload {
datasetDownload := make([]models.DatasetDownload, 0)
func GetCloudBrainDataSetInfo(uuid string, datasetname string, isNeedDown bool) []*models.DatasetDownload {
datasetDownload := make([]*models.DatasetDownload, 0)
if len(uuid) == 0 {
return datasetDownload
}
@@ -399,7 +399,7 @@ func GetCloudBrainDataSetInfo(uuid string, datasetname string, isNeedDown bool)
}
}

datasetDownload = append(datasetDownload, models.DatasetDownload{
datasetDownload = append(datasetDownload, &models.DatasetDownload{
DatasetName: name,
DatasetDownloadLink: url,
RepositoryLink: link,
@@ -1860,7 +1860,7 @@ func TrainJobShow(ctx *context.Context) {
return
}
ctx.Data["canNewJob"] = canNewJob
datasetList := make([][]models.DatasetDownload, 0)
datasetList := make([][]*models.DatasetDownload, 0)
//将运行参数转化为epoch_size = 3, device_target = Ascend的格式
for i, task := range VersionListTasks {



Loading…
Cancel
Save