From 47f20f65478a49dd157ed15b3b050c580ce59569 Mon Sep 17 00:00:00 2001 From: luoyang Date: Mon, 21 Dec 2020 11:38:02 +0800 Subject: [PATCH] Fix copy _batch_size of dataset --- mindspore/dataset/engine/datasets.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 2eb232d6b7..9301d71d7d 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -1359,6 +1359,9 @@ class Dataset: def input_indexs(self, value): self._input_indexs = value + def copy_batch_size(self, value): + self._batch_size = value + def _init_tree_getters(self): """ Get pipeline information. @@ -1931,6 +1934,7 @@ class BatchDataset(Dataset): new_op.saved_output_types = self.saved_output_types new_op.saved_output_shapes = self.saved_output_shapes new_op.input_indexs = copy.deepcopy(self._input_indexs, memodict) + new_op.copy_batch_size(copy.deepcopy(self._batch_size, memodict)) new_op.dataset_size = self.dataset_size new_op.pad = self.pad new_op.python_multiprocessing = copy.deepcopy(self.python_multiprocessing, memodict)