Imported from https://github.com/J911/WRN-pytorch