search

tfjgeorge / nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
https://nngeometry.readthedocs.io
MIT License
206 stars 20 forks source link

Import error in master #39

Closed TLESORT closed 2 years ago

TLESORT commented 2 years ago

image

tfjgeorge commented 2 years ago

Thanks for the report!

This is solved in https://github.com/tfjgeorge/nngeometry/commit/4510c476051c19d2729d118dbdda4afbf04fcd93