Browse Source

!10674 change do_copy into True in non sink_mode

From: @ms_yan
Reviewed-by: @heleiwang,@pandoublefeng
Signed-off-by: @pandoublefeng
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
4278813832
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/dataset_helper.py

+ 1
- 1
mindspore/train/dataset_helper.py View File

@@ -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


Loading…
Cancel
Save