I think that I get pretty good at cifar-10, now it is time to implement this on CUDA