Modified from https://github.com/AntixK/PyTorch-Model-Compare/tree/main
For the pytorch implementation, please refer to the above repo.
This repo implements CKA similarity for comparing deep models or datasets.
The main class for CKA computation is in cka_class.py.