# The code is based on https://github.com/pytorch/examples/blob/master/mnist.

python main.py --train-part-size 10000 --batch-size 10000 --test-part-size 1000 --test-batch-size 1000 --epochs 100 --lr 0.2 --log-interval 10

