tune the paramters to make it faster on cifar-10 (still no way near alex's implementation though)