<Refer to this for more info: https://github.com/mapillary/inplace_abn/blob/master/train_imagenet.py>