Imported from https://github.com/AlexandrosFerles/Wide-Residual-Networks-PyTorch