From c956de332792f333c1cf0a17ed92f43c3dab9e02 Mon Sep 17 00:00:00 2001 From: luoyang Date: Tue, 22 Dec 2020 19:11:08 +0800 Subject: [PATCH] optimize set_num_rows log --- .../dataset/engine/datasetops/source/sampler/sampler.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/sampler.cc b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/sampler.cc index 06accbd5cc..4534294ee3 100644 --- a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/sampler.cc +++ b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/sampler.cc @@ -142,7 +142,9 @@ int64_t SamplerRT::CalculateNumSamples(int64_t num_rows) { } Status SamplerRT::SetNumRowsInDataset(int64_t num_rows) { - CHECK_FAIL_RETURN_UNEXPECTED(num_rows > 0, "Invalid parameter, num_rows must be greater than 0."); + CHECK_FAIL_RETURN_UNEXPECTED( + num_rows > 0, + "Invalid data, data rows of input dataset must not be less than or equal to 0, please check the input dataset."); num_rows_ = num_rows; return Status::OK(); }