NGC | Catalog
CatalogModelsResNet50 v1.5 MXNet checkpoint (AMP)

ResNet50 v1.5 MXNet checkpoint (AMP)

For downloads and more information, please view on a desktop device.
Logo for ResNet50 v1.5 MXNet checkpoint (AMP)

Description

ResNet50 v1.5 MXNet checkpoint trained on Imagenet with AMP

Publisher

NVIDIA Deep Learning Examples

Latest Version

20.12.0

Modified

April 4, 2023

Size

49.08 MB

Model Overview

With modified architecture and initialization this ResNet50 version gives ~0.5% better accuracy than original.

Model Architecture

The model architecture was present in Deep Residual Learning for Image Recognition paper. The main advantage of the model is the usage of residual layers as a building block that helps with gradient propagation during training.

ResidualLayer

Image source: Deep Residual Learning for Image Recognition

Training

This model was trained using script available on NGC and in GitHub repo

Dataset

The following datasets were used to train this model:

  • ImageNet - Image database organized according to the WordNet hierarchy, in which each noun is depicted by hundreds and thousands of images.

Performance

Performance numbers for this model are available in NGC

References

License

This model was trained using open-source software available in Deep Learning Examples repository. For terms of use, please refer to the license of the script and the datasets the model was derived from.