numpee/CKA.pytorch
A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.
Jupyter NotebookApache-2.0
Issues
- 5
- 0
- 2
Support for transformer-based model.
#1 opened by geekinglcq
A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.
Jupyter NotebookApache-2.0