From cbaaffcefd8ff75d2df4df4b710298520ffaa236 Mon Sep 17 00:00:00 2001 From: ms_yan <6576637+ms_yan@user.noreply.gitee.com> Date: Fri, 27 Nov 2020 22:15:52 +0800 Subject: [PATCH] check dataset size before training --- mindspore/train/model.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 5cd2ab522f..4f6504e595 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -555,10 +555,11 @@ class Model: """ dataset_sink_mode = Validator.check_bool(dataset_sink_mode) Validator.check_is_int(sink_size) + dataset_size = train_dataset.get_dataset_size() + if dataset_size == 0: + raise ValueError("There is no valid data in dataset, please check dataset file first.") if sink_size == -1: - sink_size = train_dataset.get_dataset_size() - if sink_size == 0: - raise ValueError("There is no valid data in dataset, please check dataset file first.") + sink_size = dataset_size if sink_size < -1 or sink_size == 0: raise ValueError("The sink_size must be -1 or positive, but got sink_size {}.".format(sink_size))