tfmot.sparsity.keras.PolynomialDecay

View source on GitHub

Pruning Schedule with a PolynomialDecay function.

Inherits From: PruningSchedule

Used in the notebooks

Used in the guide

initial_sparsity Sparsity (%) at which pruning begins.
final_sparsity Sparsity (%) at which pruning ends.
begin_step Step at which to begin pruning.
end_step Step at which to end pruning.
power Exponent to be used in the sparsity function.
frequency Only apply pruning every frequency steps.

Methods

from_config

View source

Instantiates a PruningSchedule from its config.

Args
config Output of get_config().

Returns
A PruningSchedule instance.

get_config

View source

__call__

View source

Returns the sparsity(%) to be applied.

If the returned sparsity(%) is 0, pruning is ignored for the step.

Args
step Current step in graph execution.

Returns
Sparsity (%) that should be applied to the weights for the step.