Can you use the new Hessian function introduced in 1.5? https://pytorch.org/docs/stable/autograd.html#torch.autograd.functional.hessian
Can you use the new Hessian function introduced in 1.5? https://pytorch.org/docs/stable/autograd.html#torch.autograd.functional.hessian