pepy.tech

pytorch-cka

0.1.3
1.20k

Centered Kernel Alignment (CKA) with Efficient Computation and Layer-wise Visualization for PyTorch