Imported from https://github.com/stormraiser/gan-weightnorm-resnet