Code which trains a stochastic network on MNIST and computes a PAC-Bayes bound on the error rate. Requirements pytorch torchvision numpy cvxpy