首页 > 解决方案 > 如何拥有一个从多个数据集中获取批次的数据加载器,但每个批次只能包含来自一个数据集的元素?

问题描述

我需要对多个数据集执行训练。我相信最简单的方法就是 ConcatDataset,但这样我不能保证批次不会包含来自所有数据集的元素。实现它的最简单方法是什么?

标签: pytorchdatasetdataloader

解决方案


推荐阅读