Made the label_count small, now it works for cifar100 and it's even faster