NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
SSD v1.1 for PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
SSD v1.1 for PyTorch

With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.

Changelog

May 2021

  • upgrade the PyTorch container to 21.05
  • replaced APEX AMP with native PyTorch AMP
  • updated nv-cocoapi from 0.4.0 to 0.6.0
  • code updated to use DALI 1.2.0

April 2021

  • upgrade the PyTorch container to 21.04
  • changed python package naming

March 2021

  • upgrade the PyTorch container to 21.03
  • code updated to use DALI 0.30.0
  • use DALI BoxEncoder instead of a CUDA extension
  • replaced cocoapi with nv-cocoapi

June 2020

  • upgrade the PyTorch container to 20.06
  • update performance tables to include A100 results
  • update examples with A100 configs

August 2019

  • upgrade the PyTorch container to 19.08
  • update Results section in the README
  • code updated to use DALI 0.12.0
  • checkpoint loading fix
  • fixed links in the README

July 2019

  • script and notebook for inference
  • use AMP instead of hand-crafted FP16 support
  • README update
  • introduced a parameter with a path to the custom backbone checkpoint
  • minor enchantments of example/* scripts
  • alignment to changes in PyTorch 19.06

March 2019

  • Initial release