An error encountered when running code

tfjgeorge / nngeometry

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

https://nngeometry.readthedocs.io

MIT License

203 stars 20 forks source link

Closed zcb1729 closed 2 years ago

zcb1729 commented 2 years ago

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?

zcb1729 commented 2 years ago

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

tfjgeorge commented 2 years ago

You might have installed an old version of NNGeometry ?