From 20c0074d937354cfad252d3a8631966066f49f76 Mon Sep 17 00:00:00 2001 From: ms_yan <6576637+ms_yan@user.noreply.gitee.com> Date: Tue, 24 Nov 2020 20:31:42 +0800 Subject: [PATCH] raise error when dataset is zero --- mindspore/train/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 8d41452ea5..5cd2ab522f 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -554,9 +554,11 @@ class Model: >>> model.train(2, dataset) """ dataset_sink_mode = Validator.check_bool(dataset_sink_mode) + Validator.check_is_int(sink_size) if sink_size == -1: sink_size = train_dataset.get_dataset_size() - Validator.check_is_int(sink_size) + if sink_size == 0: + raise ValueError("There is no valid data in dataset, please check dataset file first.") 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))