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

Connected

Exercise

Apply dynamic quantization

You've successfully trained a neural network model for deployment, and now you want to optimize it using dynamic quantization. This step is crucial for deploying your model efficiently in environments with limited resources.

The model has been pre-loaded.

Instructions

100 XP
  • Import the necessary quantization module from PyTorch.
  • Apply dynamic quantization targeting linear layers, using 8-bit integer precision.