`File ~\Python\PyTorch\RL\utils\optim.py:61, in Sophia.hutchinson(self, p, grad)
59 def hutchinson(self, p, grad):
60 u = torch.randn_like(grad)
---> 61 hessian_vector_product = torch.autograd.grad(grad.dot(u), p, retain_graph=True)[0]
62 return u * hessian_vector_product
RuntimeError: 1D tensors expected, but got 4D and 4D tensors`
Does it run on any network architecture?
Pay now to fund the work behind this issue.
Get updates on progress being made.
Maintainer is rewarded once the issue is completed.
You're funding impactful open source efforts
You want to contribute to this effort
You want to get funding like this too