Skip to content

Commit 3a278e5

Browse files
alanchiaotensorflower-gardener
authored andcommitted
Fix broken pointer to PrunableLayer (to handle custom Keras layers) and block sparsity configuration.
PiperOrigin-RevId: 279779719
1 parent 3b820c8 commit 3a278e5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_model_optimization/g3doc/guide/pruning/train_sparse_models.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,12 +148,12 @@ tf.keras.model.save_model(final_model, file, include_optimizer=False)
148148
The pruning wrapper can also be applied to a user-defined keras layer. Custom
149149
layers can inherit from the PrunableLayer interface and implement the
150150
get_prunable_weights() method to be pruned. Please refer to
151-
[PrunableLayer](../api_docs/python/tfmot/sparsity/keras/PrunableLayer).
151+
[PrunableLayer](../../api_docs/python/tfmot/sparsity/keras/PrunableLayer).
152152

153153
### Block sparsity
154154

155155
Configure this via
156-
[prune_low_magnitude](../api_docs/python/tfmot/sparsity/keras/prune_low_magnitude).
156+
[prune_low_magnitude](../../api_docs/python/tfmot/sparsity/keras/prune_low_magnitude).
157157

158158
For some hardware architectures, it may be beneficial to induce spatially
159159
correlated sparsity. To train models in which the weight tensors have block

0 commit comments

Comments
 (0)