1. सीखें
    2. /
  2. पाठ्यक्रम
    3. /
  3. Scalable AI Models with PyTorch Lightning
Connected

अभ्यास

Creating a train DataLoader

Now that we have split our dataset, we need to define a data loader to provide batches of data during training. DataLoader efficiently loads data into memory and allows shuffling for better generalization. In this exercise, you'll complete the train_dataloader method.

निर्देश

100 XP
  • Import the DataLoader.
  • Return a DataLoader that loads self.train_data, enabling shuffling for better generalization.