Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!3652 add epoch_num description 

Merge pull request !3652 from panfengfeng/add_epoch_num_description
This commit is contained in:
mindspore-ci-bot 2020-07-29 19:33:36 +08:00 committed by Gitee
parent 68128f87a9 8803c6258d
commit 6b57b4f0e1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/train/dataset_helper.py
View File

@ -52,6 +52,7 @@ class DatasetHelper:
sink_size (int): Control the amount of data each sink.
If sink_size=-1, sink the complete dataset each epoch.
If sink_size>0, sink sink_size data each epoch. Default: -1.
epoch_num (int): Control the number of epoch data to send.
Examples:
>>> dataset_helper = DatasetHelper(dataset)