Is CCA or CKA available for captum?

  
      github.com/pytorch/captum
  
  
    
  
	  
  

  
    
      CCA/CKA support 18
    

    
      
        opened 
        
          
        
        Dec 24, 2020
      
      


      
        
          
          brando90
        
      
    
  



  Is CCA or CKA available for captum?





  
  
    
    
  
  



  
    
    
    What is a good cca, cka library for pytorch that works (ideally with GPU)? 
  
  
    I found: 


but it doesn’t seem to work reliably (or is it documented). 
Is there a standard library pytorch peeps use for cca, cka, neural net comparison analysis that actually works? (ideally on gpu)