I get this error when I use the AdaHessian. Is it a bug?

amirgholami / adahessian

ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning

MIT License

265 stars 49 forks source link

I get this error when I use the AdaHessian. Is it a bug? #33

Open ziming-liu opened 1 year ago

ziming-liu commented 1 year ago

torch_optimizer/adahessian.py", line 128, in get_trace hutchinson_trace.append(tmp_output) UnboundLocalError: local variable 'tmp_output' referenced before assignment

akashiroh commented 3 months ago

me too :(