tfjgeorge / nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

Home Page:https://nngeometry.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

An error encountered when running code

zcb1729 opened this issue · comments

I'm a rookie in python. When running the exmaple from
https://github.com/tfjgeorge/nngeometry-examples/blob/main/Gram_matrix_example.ipynb
I recieved an error
AttributeError: 'Jacobian' object has no attribute 'grads'
and I'm helpless to solve it by myself.
Do you have any ideas to solve this problem?

I ran the code on another computer with the same environment and it works well...There might be a problem with my computer.

You might have installed an old version of NNGeometry ?

T