CKA.pytorch icon indicating copy to clipboard operation
CKA.pytorch copied to clipboard

verified publisher icon numpee
→

Metadata

A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.

Results 2 CKA.pytorch issues
Sort by recently updated
recently updated
newest added

Vi

how can I print CKA of ViT and ResNet50 layer by layer

ratom avatar ratom

NaNs

5 comment

Hi there, thanks for providing the code for the CKA analysis. I have tried implementing this on my model, however, I keep on getting NaNs in the final output matrix....

repers avatar repers

← Metadata

15
Stars
0
Forks
Watchers

Owner

verified publisher icon numpee

Metadata

A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.

Back