问题描述:
【功能模块】
【操作步骤&问题现象】
1、在自定义Dataset中,按照例子IterGeneratorDataset,要是self.__data类型为list(array),数据就可以按照create_dict_iterator()输出,如果self.__data类型为list(Tensor)就不行,会出错,要不就是一直运行不结束
2、
【截图信息】
【日志信息】(可选,上传日志内容或者附件)
解决方案:
-
我们目前的设计__next__输出的数据是tuple(ndarry, ndarray, ...)
-
你现在用tensor是报什么错呢?方便提供下你的脚本吗?我们优化下处理逻辑/报错更明确些。