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()