1. 学ぶ
    2. /
  2. コース
    3. /
  3. Scalable AI Models with PyTorch Lightning
Connected

演習

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.

指示

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