ResNet50

Model description

Residual Networks, or ResNets, learn residual functions with reference to the layer inputs, instead of learning unreferenced functions. Instead of hoping each few stacked layers directly fit a desired underlying mapping, residual nets let these layers fit a residual mapping.

Prepare

Install packages

pip3 install absl-py git+https://github.com/NVIDIA/dllogger#egg=dllogger

Download datasets

Downloading and converting to TFRecord format or here make a file named imagenet_tfrecord, and store imagenet datasest convert to imagenet_tfrecord

Training

Training on single card

bash run_train_resnet50_imagenette.sh

Training on mutil-cards

bash run_train_resnet50_multigpu_imagenette.sh

Result