README.md

AnalyticFisher

Obtain Fisher Information Matrices for PyTorch Models

How to Use

Load in a PyTorch model and train the model and pass through the model, trainloader and name of the layer you wish to analyse to CalcFIM and you're good to go!