pruned base mnist

Example Script: mnist_pruning_baseline.py This script establishes a baseline model for pruning experiments using the MNIST dataset through TensorFlow Model Optimization Toolkit.

This example includes

• Training baseline model on MNIST dataset
• Performance evaluation metrics

Authors

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

Version Info

• 06/01/2024: Initial version