info-bottleneck-tracking

Code for tracking activations/gradients for pytorch models, and for computing the MI-plane

Installing dependencies

running

{conda create -n foo -c defaults -c pytorch -c conda-forge -c pyg pyg}

should install all pytorch/ pytorch_geometric dependencies.

{pip install ogb}

gets the datasets from the open-graph benchmark ()