change do_copy into True in non sink_mode

4 years ago · 819aea50ad
parent 9366200dba
commit 819aea50ad
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/dataset_helper.py
+++ b/mindspore/train/dataset_helper.py
@ -395,7 +395,7 @@ class _DatasetIterNormal:
        self.dataset = dataset
        self.device_num = _get_device_num()
        self.global_rank = _get_global_rank()
-        self.iter = self.dataset.create_tuple_iterator(num_epochs=epoch_num, do_copy=False)
+        self.iter = self.dataset.create_tuple_iterator(num_epochs=epoch_num, do_copy=True)

    def __iter__(self):
        return self