This code was used, in part, for our paper: Leveraging the Graph Structure of Neural Network Training Dynamics. Fatemeh Vahedian, Ruiyu Li, Puja Trivedi, Di Jin, Danai Koutra. Short Paper, CIKM 2022.
- numpy
- scipy
- networkx
- torch
- torchvision
- Cython (optional)
- Ripser (optional)
This packages builds off of several great projects including, Neural Network Graph Homology Package. Thanks!