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

check dataset size before training

This commit is contained in:
ms_yan 2020-11-27 22:15:52 +08:00
parent 47d854143f
commit cbaaffcefd
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))