如何解决具有IterableDataset的Dataloder引发异常:预期的未指定batch_sampler选项
我尝试通过IteratableDataset
实现自定义Dataloder
,如下所示:
class LazyDataset(IterableDataset):
def __init__(self,file_dir):
self.file_dir = file_dir
self.files = os.listdir(file_dir)
def __len__(self) -> int:
return len(self.files)
def __iter__(self):
for file in self.files:
# read and preprocess file
yield data
train_dataset = LazyDataset('/train')
train_loader = DataLoader(train_dataset,batch_size=args.train_batch_size,num_workers=args.n_workers
)
它工作正常,但是当我与ignite-engine集成时,它会引发以下异常:
"batch_sampler option,but got batch_sampler={}".format(batch_sampler))
ValueError: DataLoader with IterableDataset: expected unspecified batch_sampler option,but got batch_sampler=<ignite.engine.engine.ReproducibleBatchSampler
如何解决以上问题?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。