From 4644085e9288233718b693f83fe4478bc7a9d8a3 Mon Sep 17 00:00:00 2001 From: panfengfeng Date: Wed, 29 Jul 2020 16:10:06 +0800 Subject: [PATCH] add epoch_num --- mindspore/train/dataset_helper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py index 0c6e7eb4648..00d0267011f 100644 --- a/mindspore/train/dataset_helper.py +++ b/mindspore/train/dataset_helper.py @@ -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)