小编Joh*_*ton的帖子

如何通过单个操作合并两个 torch.utils.data 数据加载器

我有两个数据加载器,我想合并它们而不重新定义数据集,在我的例子中是 train_dataset 和 val_dataset。

train_loader = DataLoader(train_dataset, batch_size = 512, drop_last=True,shuffle=True)
val_loader = DataLoader(val_dataset, batch_size = 512, drop_last=False)
Run Code Online (Sandbox Code Playgroud)

想要的结果:

train_loader = train_loader + val_loader 
Run Code Online (Sandbox Code Playgroud)

deep-learning pytorch dataloader

4
推荐指数
2
解决办法
1万
查看次数

标签 统计

dataloader ×1

deep-learning ×1

pytorch ×1