fix padded_sample default

1909d265 · liyong · ca94916e · 1909d265
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

mindspore/dataset/engine/datasets.py mindspore/dataset/engine/datasets.py +2 -1

未找到文件。
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -2746,8 +2746,9 @@ class MindDataset(MappableDataset):

    def get_args(self):
        args = super().get_args()
-        padded_sample = {}
+        padded_sample = None
        if self.padded_sample:
+            padded_sample = {}
            for k, v in self.padded_sample.items():
                if isinstance(v, np.ndarray):
                    padded_sample[k] = v.tobytes()