提交 c3b1814d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2224 [MD] modify padded_sample default value

Merge pull request !2224 from liyong126/mindrecord_fix_mi
......@@ -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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册