Python package for extracting representations from state-of-the-art computer vision models
torch>=2.0.0 and torchvision=0.15.* (fixes issue #165)get_cka where the user can set the backend (torch or numpy); function returns the CKA objectthingsvision model collectionwith statement for PyTorch models) for custom data pipelinethingsvision