pepy.tech

pytorch-cka

1.0.1
3.07k

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