From 732dd01a6aa88e855079ebfbced6abae1a36d262 Mon Sep 17 00:00:00 2001 From: Yanjun Peng Date: Mon, 13 Apr 2020 17:46:48 +0800 Subject: [PATCH] fix epoch repeat doc problem --- mindspore/dataset/engine/datasets.py | 2 ++ mindspore/train/model.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 3d660d58a8..4e0b082734 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -394,6 +394,8 @@ class Dataset: The order of using repeat and batch reflects the number of batches. Recommend that repeat operation should be used after batch operation. If dataset_sink_mode is False, here repeat operation is invalid. + If dataset_sink_mode is True, repeat count should be euqal to the epoch of training. Otherwise, + errors could occur since the amount of data is not the amount training requires. Args: count (int): Number of times the dataset should be repeated (default=None). diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 833fb07256..55db3b538a 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -360,6 +360,9 @@ class Model: Note: CPU is not supported when dataset_sink_mode is true. + If dataset_sink_mode is True, epoch of training should be equal to the count of repeat + operation in dataset processing. Otherwise, errors could occur since the amount of data + is not the amount training requires. Args: epoch (int): Total number of iterations on the data.