xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9131 dataset: check dataset size first before training 

From: @ms_yan
Reviewed-by: @pandoublefeng,@heleiwang
Signed-off-by: @pandoublefeng
This commit is contained in:
mindspore-ci-bot 2020-11-28 11:46:46 +08:00 committed by Gitee
parent 6633c6cf19 cbaaffcefd
commit a83c1053cd
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/train/model.py
View File

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