tf.contrib.model_pruning.Pruning

View source on GitHub

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

Adds summaries of weight sparsities and thresholds.

conditional_mask_update_op

View source

mask_update_op

View source

View source