当我有不同的长度数据集时，如何为PyTorch数据加载器定义__len__方法？

import torch
from torch.utils.data import Dataset, DataLoader

class CustomDataset(Dataset):
    def __init__(self, data):
        self.data = data

    def __len__(self):
        return len(self.data)

    def __getitem__(self, index):
        return self.data[index]

# 创建数据集
data = [1, 2, 3, 4, 5]
dataset = CustomDataset(data)

# 创建数据加载器
dataloader = DataLoader(dataset, batch_size=2, shuffle=True)

# 打印数据集长度
print(len(dataset))

# 打印数据加载器长度
print(len(dataloader))