Skip to content

pruned mnist

Example Script: pruned_mnist.py

This script demonstrates the application of model pruning on the MNIST dataset using TensorFlow Model Optimization Toolkit. The pruning process reduces model size while maintaining accuracy.

This example includes
  • Loading and preprocessing MNIST dataset
  • Implementing model pruning for compression
  • Training and evaluating pruned models
Authors
  • Nithyashree R (nithyashreer@iisc.ac.in)
Version Info
  • 06/01/2024: Initial version