tensorflow / tensorflow/model-optimization

Polynomial decay for pruning schedule doesn't prune the weight

Open
#601 1 comment 0 reactions 1 assignee View on GitHub

@Xhark is already working on this.

Since Apr 5, 2021.

bug
Dominant language
Python
Stars
1.6k
Forks
349
Avg merge
3d 2h
Merged PRs (30d)
1

Description

tfmot.sparsity.keras.PolynomialDecay can be run but the weight is still non-zero. When I change it to tfmot.sparsity.keras.ConstantSparsity it just works fine. Is there any advice?
I prune specific layer (only conv2D) if that helps.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.