Missing property in HessianTracker - Githubissues

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

MIT License

360 stars 43 forks source link

Missing property in HessianTracker #15

Closed themightyoarfish closed 5 years ago

themightyoarfish commented 5 years ago

https://github.com/noahgolmant/pytorch-hessian-eigenthings/blob/12741ff90c451214fb83913da2bf93fbacda20b0/hessian_eigenthings/tracking.py#L72

I think you forgot to init self.power_iter_steps in the initialiser.