Resource
With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.
Use the NGC CLI to download:
Copied!
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