Browse Source

!8968 dataset: raise error when dataset size is zero

From: @ms_yan
Reviewed-by: @liucunwei,@pandoublefeng
Signed-off-by: @liucunwei
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c65ce4f32f
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/train/model.py

+ 3
- 1
mindspore/train/model.py View File

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



Loading…
Cancel
Save