From dafe76552801372821c99eaf5c4321fbb8919c16 Mon Sep 17 00:00:00 2001 From: Mathieu Blondel Date: Fri, 10 Jun 2022 18:31:29 +0200 Subject: [PATCH] Use max_stepsize in init. --- jaxopt/_src/lbfgs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxopt/_src/lbfgs.py b/jaxopt/_src/lbfgs.py index b34339a7..10fc3c81 100644 --- a/jaxopt/_src/lbfgs.py +++ b/jaxopt/_src/lbfgs.py @@ -228,7 +228,7 @@ def init_state(self, """ return LbfgsState(iter_num=jnp.asarray(0), value=jnp.asarray(jnp.inf), - stepsize=jnp.asarray(1.0), + stepsize=jnp.asarray(self.max_stepsize), error=jnp.asarray(jnp.inf), s_history=init_history(init_params, self.history_size), y_history=init_history(init_params, self.history_size),