tfmot.sparsity.keras.UpdatePruningStep

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

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