Attend the Women in ML Symposium on December 7 Register now

tfmot.sparsity.keras.PruningSummaries

Stay organized with collections Save and categorize content based on your preferences.

A Keras callback for adding pruning summaries to tensorboard.

Used in the notebooks

Used in the guide

Logs the sparsity(%) and threshold at a given iteration step.

Methods

set_model

Sets Keras model and writes graph if specified.

set_params