GDIteration
- class deepinv.optim.optim_iterators.GDIteration(**kwargs)[source]
Bases:
OptimIterator
Iterator for Gradient Descent.
Class for a single iteration of the gradient descent (GD) algorithm for minimising \(f(x) + \lambda g(x)\).
The iteration is given by
\[\begin{split}\begin{equation*} \begin{aligned} v_{k} &= \nabla f(x_k) + \nabla g(x_k) \\ x_{k+1} &= x_k-\gamma v_{k} \end{aligned} \end{equation*}\end{split}\]where \(\gamma\) is a stepsize.
- forward(X, cur_data_fidelity, cur_prior, cur_params, y, physics)[source]
Single gradient descent iteration on the objective \(f(x) + \lambda g(x)\).
- Parameters:
X (dict) – Dictionary containing the current iterate \(x_k\).
cur_data_fidelity (deepinv.optim.DataFidelity) – Instance of the DataFidelity class defining the current data_fidelity.
cur_prior (deepinv.optim.prior) – Instance of the Prior class defining the current prior.
cur_params (dict) – Dictionary containing the current parameters of the algorithm.
y (torch.Tensor) – Input data.
- Returns:
Dictionary {“est”: (x, ), “cost”: F} containing the updated current iterate and the estimated current cost.