Resource
With modified architecture and initialization this ResNet50 version gives ~0.5% better accuracy than original.
Use the NGC CLI to download:
Copied!
Changelog
- March, 2019
- Initial release
- May, 2019
- Added DALI support
- Added scripts for DGX-2
- Added benchmark results for DGX-2 and XLA-enabled DGX-1 and DGX-2.
- July, 2019
- Added Cosine learning rate schedule
- August, 2019
- Added mixup regularization
- Added T4 benchmarks
- Improved inference capabilities
- Added SavedModel export
- January, 2020
- Removed manual checks for dataset paths to facilitate cloud storage solutions
- Move to a new logging solution
- Bump base docker image version
- March, 2020
- Code cleanup and refactor
- Improved training process
- June, 2020
- Added benchmark results for DGX-A100
- Updated benchmark results for DGX-1, DGX-2 and T4
- Updated base docker image version
- August 2020
- Updated command line argument names
- Added support for syntetic dataset with different image size
- January, 2022
- Added barrier at the end of multiprocess run
Known issues
Performance without XLA enabled is low due to BN + ReLU fusion bug.