An Numpy and PyTorch Implementation of CKA-similarity with CUDA support
Primary LanguageJupyter Notebook