libreasr.lib.optimizer.AdaHessian

libreasr.lib.optimizer.AdaHessian(params, lr=0.1, hessian_power=1, hutchinson_trace=None, mom=0.9, sqr_mom=0.98, eps=0.0001, wd=0.0, decouple_wd=True)[source]

A Optimizer for Adam with lr, mom, sqr_mom, eps and params