numpee / CKA.pytorch

A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.
34Updated 8 months ago

Related projects: