Imported from https://github.com/mf1024/Batch-Renormalization-PyTorch