Use max_stepsize in init.

google · Jun 10, 2022 · dafe765 · dafe765
1 parent b265542
commit dafe765
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),