/Pytorch_Mnist_BN

Implementing BN on MNIST dataset with Pytorch

Primary LanguagePython

Pytorch_Mnist_BN

Implementing BN on MNIST dataset with Pytorch. You can carry out mutiple experiments through a change of parameters and observe the benefit of BN through the two pre-set networks(with and without Batch Normalization).

python mnist_bn.py