NVIDIA Deep Learning Examples ResNet50 Model

In this example, we show how to optimize the ResNet50 model from the Deep Learning Examples repository. We recommend running this example in NVIDIA NGC PyTorch container.

Install dependencies and download the model repository and the dataset

. ./install.sh

This script downloads the model repository and dummy version of ImageNet dataset. To use the original ImageNet dataset on which the model was trained on, please follow instructions from the model repository and in the next step specify the dataset directory with --data-path argument.

Run optimization

./optimize.py