model-optimization icon indicating copy to clipboard operation
model-optimization copied to clipboard

pruning example code for GNMT

Open TaeheeJeong opened this issue 4 years ago • 1 comments

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.

TaeheeJeong avatar Aug 04 '21 23:08 TaeheeJeong

Sorry, the example code is not available at this moment.

Btw, I cannot imagine any specific issue on applying Pruning API to GNMT model, so what about trying it similar to the guide in https://www.tensorflow.org/model_optimization/guide/pruning/pruning_with_keras ? (but with GNMT model)

If you need any further assistance, please let me know.

rino20 avatar Sep 28 '21 14:09 rino20