拆分数据加载器PyTorch

Question

是否可以将训练数据集的数据加载器对象拆分为训练和验证数据加载器？

from torch.utils.data import DataLoader
from torchvision import datasets, transforms
train_dataset = datasets.ImageFolder(train_data_directory, transform=transforms.ToTensor())
# Data loader
train_loader = DataLoader(train_dataset, batch_size=100, shuffle=True)

现在，我想将一个train_loader拆分为train和validation数据加载器。

Answer 1

查看torch.utils.data中的random_split。它将处理随机的Dataset分割(必须在创建DataLoader之前进行分割，而不是在创建之后(。