From ac6d3d557c89892cf3dbbadae8eaccfdeb6a7a1e Mon Sep 17 00:00:00 2001 From: jinyaohui Date: Tue, 28 Jul 2020 09:36:54 +0800 Subject: [PATCH] add description about sink_size --- mindspore/train/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 844480d20d8..842d23221f0 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -523,7 +523,9 @@ class Model: of data will be transferred one by one. The limitation of data transmission per time is 256M. Args: - epoch (int): Total number of iterations on the data. + epoch (int): Generally, total number of iterations on the data per epoch. + When dataset_sink_mode is set to true and sink_size>0, each epoch sink sink_size + steps on the data instead of total number of iterations. train_dataset (Dataset): A training dataset iterator. If there is no loss_fn, a tuple with multiply data (data1, data2, data3, ...) should be returned and passed to the network. Otherwise, a tuple (data, label) should