CKA.pytorch
CKA.pytorch copied to clipboard
NaNs
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. Any idea why this happens? Thanks