Running time of MNISTSuperpixels

[nukappa]

Hi, thank you very much for the package.

Could you comment on the running times? I cloned the repo and ran the train_models MNIST, it loads the dataset and starts training, but (1) training takes forever (30 cores CPU -- no GPU available) and (2) it doesn't seem to improve after a few epochs:

$ python train_models.py MNISTSuperpixels True Epoch: 001, Loss: 7.64400, Train Acc: 0.10218, Test Acc: 0.10100 Epoch: 002, Loss: 2.30219, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 003, Loss: 2.30138, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 004, Loss: 2.30126, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 005, Loss: 2.30124, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 006, Loss: 2.30124, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 007, Loss: 2.30123, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 008, Loss: 2.30124, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 009, Loss: 2.30124, Train Acc: 0.11237, Test Acc: 0.11350 Epoch: 010, Loss: 2.30124, Train Acc: 0.11237, Test Acc: 0.11350

It took ~2days for training 10 epochs. Any help is much appreciated.