How it works...

Here, we are training the network for a batch size of 10. If we increase it, the network performance decreases. Also, the accuracy of the trained network is checked on the test data; the size of the test data on which it is tested is 1,000.