Made the label_count small, now it works for cifar100 and it's even faster
Simple optimization on memory usage.