tfmot.sparsity.keras.PolynomialDecay

Pruning Schedule with a PolynomialDecay function.

tfmot.sparsity.keras.PolynomialDecay(
    initial_sparsity,
    final_sparsity,
    begin_step,
    end_step,
    power=3,
    frequency=100
)

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

Args
`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

@classmethod
from_config(
    config
)

Instantiates a PruningSchedule from its config.

Args
`config`	Output of `get_config()`.

Returns
A `PruningSchedule` instance.

get_config()

__call__(
    step
)

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.