PyTorch code for hierarchical k-means -- a data curation method for self-supervised learning - View it on GitHub