Imported from https://github.com/ZhaoJ9014/shallower-wider-ResNet-Model-A-SW-ResNet-A