Public Member Functions | |
| __init__ (self, np.ndarray x, Callable|float step, Callable|None grad=None, Callable|None prox=None, float|SingleItemArray b1=0.9, float b2=0.999, float eps=1e-8, float p=0.25, np.ndarray|None m0=None, np.ndarray|None v0=None, np.ndarray|None vhat0=None, str scheme="amsgrad", float prox_e_rel=1e-6) | |
| update (self, int it, np.ndarray input_grad, *args) | |
| AdaproxParameter | __deepcopy__ (self, dict[int, Any]|None memo=None) |
| AdaproxParameter | __copy__ (self) |
Public Member Functions inherited from lsst.scarlet.lite.parameters.Parameter | |
| __init__ (self, np.ndarray x, dict[str, np.ndarray] helpers, Callable|float step, Callable|None grad=None, Callable|None prox=None) | |
| float | step (self) |
| tuple[int,...] | shape (self) |
| npt.DTypeLike | dtype (self) |
| Parameter | __copy__ (self) |
| Parameter | __deepcopy__ (self, dict[int, Any]|None memo=None) |
| Parameter | copy (self, bool deep=False) |
| resize (self, Box old_box, Box new_box) | |
Public Attributes | |
| b1 = _b1 | |
| b2 = b2 | |
| eps = eps | |
| p = p | |
| scheme = scheme | |
| phi_psi = phi_psi[scheme] | |
| e_rel = prox_e_rel | |
Public Attributes inherited from lsst.scarlet.lite.parameters.Parameter | |
| x = x | |
| helpers = helpers | |
| grad = grad | |
| prox = prox | |
Additional Inherited Members | |
Protected Attributes inherited from lsst.scarlet.lite.parameters.Parameter | |
| _step = _step | |
Operator updated using te Proximal ADAM algorithm
Uses multiple variants of adaptive quasi-Newton gradient descent
* Adam (Kingma & Ba 2015)
* NAdam (Dozat 2016)
* AMSGrad (Reddi, Kale & Kumar 2018)
* PAdam (Chen & Gu 2018)
* AdamX (Phuong & Phong 2019)
* RAdam (Liu et al. 2019)
See details of the algorithms in the respective papers.
| AdaproxParameter lsst.scarlet.lite.parameters.AdaproxParameter.__copy__ | ( | self | ) |
Create a shallow copy of this parameter.
Returns
-------
parameter:
A shallow copy of this parameter.
| AdaproxParameter lsst.scarlet.lite.parameters.AdaproxParameter.__deepcopy__ | ( | self, | |
| dict[int, Any] | None | memo = None ) |
Create a deep copy of this parameter.
Parameters
----------
memo:
A memoization dictionary used by `copy.deepcopy`.
Returns
-------
parameter:
A deep copy of this parameter.
| lsst.scarlet.lite.parameters.AdaproxParameter.update | ( | self, | |
| int | it, | ||
| np.ndarray | input_grad, | ||
| * | args ) |
Update the parameter and meta-parameters using the PGM See `~Parameter` for more.
Reimplemented from lsst.scarlet.lite.parameters.Parameter.