Skip to content

tensorflow & keras weight pruning to compress the machine learning models for better inference time

Notifications You must be signed in to change notification settings

Asrst/tf-weight-pruning

Repository files navigation

Tensorflow model compression using Pruning

DataSet:fashion_mnist

tf_keras_pruning_fmnist.ipynb: contains code used for training, functions used for pruning, & line plot of results after pruning. (in orderly manner)

tf_keras_pruning.py: contains KerasModelPruner class.

tf_pruning_fmnist.ipynb: contains raw tensorflow implementation for training & weight pruning & results after pruning.

tensorflow/keras weight pruning

About

tensorflow & keras weight pruning to compress the machine learning models for better inference time

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published