tf.contrib.model_pruning.Pruning

tf.contrib.model_pruning.Pruning(
    spec=None, global_step=None, sparsity=None
)

Args
spec	Pruning spec as defined in pruning.proto
global_step	A tensorflow variable that is used while setting up the sparsity function
sparsity	A tensorflow scalar variable storing the sparsity

Methods

add_pruning_summaries

View source

add_pruning_summaries()

Adds summaries of weight sparsities and thresholds.

conditional_mask_update_op

View source

conditional_mask_update_op()

mask_update_op

View source

mask_update_op()

print_hparams

View source

print_hparams()