分裂DataLoader PyTorch

Question

是否有可能将训练数据集的dataloader对象拆分为训练和验证数据集？

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来训练和验证数据中心。

Answer 1

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