Number of data in each class
resnet50
resnet101
resnet18