jayroxis / CKA-similarity

An Numpy and PyTorch Implementation of CKA-similarity with CUDA support

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jayroxis/CKA-similarity Watchers