diff --git a/src/nemos/solvers/_svrg.py b/src/nemos/solvers/_svrg.py index c381ba32..8f47b8e6 100644 --- a/src/nemos/solvers/_svrg.py +++ b/src/nemos/solvers/_svrg.py @@ -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