Training function (include SVI)

LeNet (include SVI)

Utility function, including data functions

Start testing

Compare the accuracies

Note, as long as results saved on file, can directly run these below

MNIST (fully data with batches, SVI only at CNN layers)

CIFAR-10 (fully data with batches, SVI only at CNN layers)