tfmot.sparsity.keras.UpdatePruningStep

View source on GitHub

Keras callback which updates pruning wrappers with the optimizer step.

Used in the notebooks

Used in the guide

This callback must be used when training a model which needs to be pruned. Not doing so will throw an error.

Example:

model.fit(x, y,
    callbacks=[UpdatePruningStep()])

Methods

set_model

set_params