HuggingFace BERT Model

In this example, we show how to optimize BERT model from HuggingFace repository. We recommend running this example in NVIDIA NGC TensorFlow2 container.

Install dependencies

./install.sh

Run optimization

./optimize.py