Benchmark

yi@WangYis-iMac:~/go/src/github.com/wangkuiyi/gotorch/benchmark/python (mnist_benchmark)*$ time python mnist.py
Datset size: 60000
Train Epoch: 0, BatchIdx: 0, Loss: 2.316286
Train Epoch: 0, BatchIdx: 200, Loss: 0.481705
Train Epoch: 0, BatchIdx: 400, Loss: 0.309341
Train Epoch: 0, BatchIdx: 600, Loss: 0.248037
Train Epoch: 0, BatchIdx: 800, Loss: 0.369942
Train Epoch: 1, BatchIdx: 0, Loss: 0.217632
Train Epoch: 1, BatchIdx: 200, Loss: 0.195020
Train Epoch: 1, BatchIdx: 400, Loss: 0.212828
Train Epoch: 1, BatchIdx: 600, Loss: 0.168781
Train Epoch: 1, BatchIdx: 800, Loss: 0.263356
Train Epoch: 2, BatchIdx: 0, Loss: 0.152046
Train Epoch: 2, BatchIdx: 200, Loss: 0.151219
Train Epoch: 2, BatchIdx: 400, Loss: 0.171766
Train Epoch: 2, BatchIdx: 600, Loss: 0.126567
Train Epoch: 2, BatchIdx: 800, Loss: 0.191472
Train Epoch: 3, BatchIdx: 0, Loss: 0.110657
Train Epoch: 3, BatchIdx: 200, Loss: 0.125849
Train Epoch: 3, BatchIdx: 400, Loss: 0.137188
Train Epoch: 3, BatchIdx: 600, Loss: 0.101382
Train Epoch: 3, BatchIdx: 800, Loss: 0.156615
Train Epoch: 4, BatchIdx: 0, Loss: 0.082411
Train Epoch: 4, BatchIdx: 200, Loss: 0.109833
Train Epoch: 4, BatchIdx: 400, Loss: 0.109931
Train Epoch: 4, BatchIdx: 600, Loss: 0.086049
Train Epoch: 4, BatchIdx: 800, Loss: 0.138978
The throughput: 865.1952912694164 samples.sec

real	1m12.950s
user	3m47.505s
sys	0m11.465s