HuggingFace BERT Model

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

Install dependencies

./install.sh

Run optimization

./optimize.py