1. Learn
    2. /
  2. Courses
    3. /
  3. Scalable AI Models with PyTorch Lightning
Connected

Exercise

Optimizing model training with Lightning

By implementing automated techniques like ModelCheckpoint and EarlyStopping, you'll ensure your model selects the best-performing parameters while avoiding unnecessary computations.

The dataset, a subset of the Osmanya MNIST dataset, provides a real-world use case where scalable AI training techniques can significantly improve efficiency and accuracy.

OsmanyaDataModule and ImageClassifier have been predefined for you.

Instructions

100 XP
  • Import callbacks that you'll use for model checkpointing and early stopping.
  • Train the model with the ModelCheckpoint and EarlyStopping callbacks.