AttributeError: 'HVPOperator' object has no attribute 'zero_grad'

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

MIT License

360 stars 43 forks source link

AttributeError: 'HVPOperator' object has no attribute 'zero_grad' #38

Closed neerajkuhike closed 3 years ago

neerajkuhike commented 3 years ago

I am facing this issue when running resnet 18 on cifar 10 dataset.

AjinkyaBankar commented 3 years ago

Did you get the solution? I am also facing the same problem.

noahgolmant commented 3 years ago

This should be fixed now in master! The error was because I renamed zero_grad to _zero_grad in a previous PR! Please reopen if this does not fix it on your end