शुरू करेंमुफ़्त में शुरू करें

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.

यह अभ्यास पाठ्यक्रम का हिस्सा है

Scalable AI Models with PyTorch Lightning

पाठ्यक्रम देखें

अभ्यास निर्देश

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

इंटरैक्टिव व्यावहारिक अभ्यास

इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।

# Import libraries
from torch.utils.data import ____
import lightning.pytorch as pl

class LoaderDataModule(pl.LightningDataModule):
    def __init__(self):
        super().__init__()
        self.train_data = None
        self.val_data = None
    def setup(self, stage=None):
        self.train_data, self.val_data = random_split(dataset, [80, 20])
    def train_dataloader(self):
      	# Complete DataLoader
        return ____(____, batch_size=16, shuffle=____)
कोड संपादित करें और चलाएँ