Pytorch implementation of EvoNorms on CIFAR