numpee / CKA.pytorch

A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.
Apache License 2.0
35 stars 1 forks source link