NVIDIA Deep Learning Examples
ResNet50 v1.5 for TensorFlow1
Resource
NVIDIA Deep Learning Examples
ResNet50 v1.5 for TensorFlow1

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

  • Changelog

    1. March, 2019
    • Initial release
    1. May, 2019
    • Added DALI support
    • Added scripts for DGX-2
    • Added benchmark results for DGX-2 and XLA-enabled DGX-1 and DGX-2.
    1. July, 2019
    • Added Cosine learning rate schedule
    1. August, 2019
    • Added mixup regularization
    • Added T4 benchmarks
    • Improved inference capabilities
    • Added SavedModel export
    1. January, 2020
    • Removed manual checks for dataset paths to facilitate cloud storage solutions
    • Move to a new logging solution
    • Bump base docker image version
    1. March, 2020
    • Code cleanup and refactor
    • Improved training process
    1. June, 2020
    • Added benchmark results for DGX-A100
    • Updated benchmark results for DGX-1, DGX-2 and T4
    • Updated base docker image version
    1. August 2020
    • Updated command line argument names
    • Added support for syntetic dataset with different image size
    1. January, 2022
    • Added barrier at the end of multiprocess run

    Known issues

    Performance without XLA enabled is low due to BN + ReLU fusion bug.