README.md

NVIDIA Deep Learning Examples EfficientNet Model

In this example we show how to optimize EfficientNet model from Deep Learning Examples repository. We recommend running this example in NVIDIA NGC TensorFlow2 container.

Install dependencies and download model repository

. ./install.sh

Run optimization

./optimize.py --model-name EfficientNet-v1-B0