Browse Source

!2224 [MD] modify padded_sample default value

Merge pull request !2224 from liyong126/mindrecord_fix_mi
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
c3b1814d61
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/dataset/engine/datasets.py

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

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


Loading…
Cancel
Save