Skip to content

Commit

Permalink
fix comment
Browse files Browse the repository at this point in the history
  • Loading branch information
BalzaniEdoardo committed Oct 9, 2024
1 parent 4fdcf32 commit 30934ca
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/nemos/solvers/_svrg.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,7 @@ def _inner_loop_param_update_step(
# gradient of f_{i_k} at x_{k} in the pseudocode of Gower et al. 2020
minibatch_grad_at_current_params = self.loss_gradient(params, *args)
# gradient on batch_{i_k} evaluated at the anchor point
# gradient of f_{i_k} at x_{x} in the pseudocode of Gower et al. 2020
# gradient of f_{i_k} at x_{k} in the pseudocode of Gower et al. 2020
minibatch_grad_at_reference_point = self.loss_gradient(reference_point, *args)

# SVRG gradient estimate
Expand Down

0 comments on commit 30934ca

Please sign in to comment.