提交代码

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"))