PyTorch:从多个数据集进行批处理

2024-04-27 04:59:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我想在培训中使用多个数据集。我希望每个批次都来自一个数据集,但每个历元中的所有数据集(可能)都有批次

将数据集合并到一个简单的Dataset对象中并使用默认的Dataloader会导致在一个批处理中包含来自不同数据集的样本

我自己的猜测是为每个数据集都有一个单独的Dataset对象,并覆盖数据加载器或采样器,但我不知道怎么做


Tags: 数据对象dataset样本采样器dataloader历元