tensorflow / tensorflow/model-optimization

pruning example code for GNMT

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

@rino20 is already working on this.

Since Aug 9, 2021.

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

Description

System information

  • TensorFlow version (you are using): 2.5.0
  • Are you willing to contribute it (Yes/No): Yes

Motivation

There is pruning result of GNMT EN-DE and GNMT DE-EN on https://blog.tensorflow.org/2019/05/tf-model-optimization-toolkit-pruning-API.html and https://www.tensorflow.org/model_optimization/guide/pruning.

However, I could not find any example code how to prune GNMT EN-DE or DE-EN.

Could you someone upload example code (tutorial) how to prune GNMT using pruning API?

I am interested in pruning Transformer and BERT models.

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.