Obtain Fisher Information Matrices for PyTorch Models
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!
-
Notifications
You must be signed in to change notification settings - Fork 0
MattBromley98/AnalyticFisher
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Obtain Fisher Information Matrices for PyTorch Models
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published