tfmot.sparsity.keras.PruningSummaries

A Keras callback for adding pruning summaries to tensorboard.

tfmot.sparsity.keras.PruningSummaries(
    log_dir, update_freq='epoch', **kwargs
)

Used in the notebooks

Used in the guide
Pruning comprehensive guide Pruning for on-device inference w/ XNNPACK Pruning in Keras example

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

Methods

set_model

set_model(
    model
)

Sets Keras model and writes graph if specified.

set_params

set_params(
    params
)