How do I load data in PyTorch?

How do I load data in PyTorch?

You can use the torch.utils.data module to load data in PyTorch. Do you have a specific dataset in mind?

I'm working with a custom dataset. How can I create a DataLoader for it?

You can create a custom dataset by subclassing torch.utils.data.Dataset and implementing the __len__ and __getitem__ methods. Then, use DataLoader to load it.

Can you show me an example of a custom dataset?

Sure! Here's a simple example:

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

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

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

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

# Usage
my_data = MyDataset(data=[1, 2, 3, 4])
dataloader = DataLoader(my_data, batch_size=2, shuffle=True)