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

अभ्यास

Splitting data with LightningDataModule

You will complete the setup method in a LightningDataModule. Proper dataset partitioning ensures that the model is trained on one subset and validated on another, preventing overfitting.

The dataset has already been pre-imported.

निर्देश

100 XP
  • Import random_split to split the dataset into training and validation.
  • Split the dataset into training (80%) and validation (20%) using random_split.