pepy.tech

pytorch-cka

1.0.1
2.55k

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