Change test DataLoader to use the test batch size

The batch size for the training set was erroneously used for instantiating the DataLoader class for the test set.

Author: rdinse

mnist/main.py

@@ -46,7 +46,7 @@
  transforms.ToTensor(),
  transforms.Normalize((0.1307,), (0.3081,))
  ])),
- batch_size=args.batch_size, shuffle=True, **kwargs)
+ batch_size=args.test_batch_size, shuffle=True, **kwargs)
 
 
 class Net(nn.Module):