Hi, just wondering if anyone has PyTorch code for dividing the CIFAR-10 dataset into Head and Tail classes (for example if the count of classes is less than 3000 it should be considered tail else head to class) and then feeding these classes to CNN for feature extraction and then the angular variance of head and tail classes are found. Any help or suggestion will be highly appreciated.