Skip to content

Commit

Permalink
Pass parameter value to optax's optimizer update
Browse files Browse the repository at this point in the history
This is an optional parameter that it doesn't hurt to pass.
  • Loading branch information
NeilGirdhar authored and patrick-kidger committed Aug 18, 2024
1 parent 4fc86e0 commit bfffa9b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimistix/_solver/optax.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def step(
("loss" in self.verbose, "Loss", f),
("y" in self.verbose, "y", y),
)
updates, new_opt_state = self.optim.update(grads, state.opt_state)
updates, new_opt_state = self.optim.update(grads, state.opt_state, y)
new_y = eqx.apply_updates(y, updates)
terminate = cauchy_termination(
self.rtol,
Expand Down

0 comments on commit bfffa9b

Please sign in to comment.