From bd96fa5515c0799d84f788f7751400896aaaddd5 Mon Sep 17 00:00:00 2001 From: ychao_1983 Date: Thu, 3 Nov 2022 15:02:50 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/cloudbrain/cloudbrainTask/train.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/services/cloudbrain/cloudbrainTask/train.go b/services/cloudbrain/cloudbrainTask/train.go index b59cbc804..8e4673d66 100644 --- a/services/cloudbrain/cloudbrainTask/train.go +++ b/services/cloudbrain/cloudbrainTask/train.go @@ -530,7 +530,12 @@ func checkParameters(ctx *context.Context, option api.CreateTrainJobOption, lock var datasetInfos map[string]models.DatasetInfo var datasetNames string if option.Type != TaskTypeModelArts { - datasetInfos, datasetNames, err = models.GetDatasetInfo(option.Attachment, computeType) + if isC2NetTask(option) { + datasetInfos, datasetNames, err = models.GetDatasetInfo(option.Attachment, computeType) + } else { + datasetInfos, datasetNames, err = models.GetDatasetInfo(option.Attachment) + } + if err != nil { log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"]) return nil, nil, "", fmt.Errorf(ctx.Tr("cloudbrain.error.dataset_select"))