quantized base mnist

Example Script: quantization_base_mnist.py

This script demonstrates quantization-aware training on the MNIST dataset. It gives the performance of the baseline model.

This example includes

• Training baseline model on MNIST dataset
• Evaluating model accuracy before QAT

Authors

• Nithyashree R (nithyashreer@iisc.ac.in)

Version Info

• 06/01/2024: Initial version