lsst.scarlet.lite ge1c02a5578+b0138be388
 
Loading...
Searching...
No Matches
lsst.scarlet.lite.parameters.AdaproxParameter Class Reference
Inheritance diagram for lsst.scarlet.lite.parameters.AdaproxParameter:
lsst.scarlet.lite.parameters.Parameter

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
 

Detailed Description

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.

Member Function Documentation

◆ __copy__()

AdaproxParameter lsst.scarlet.lite.parameters.AdaproxParameter.__copy__ ( self)
Create a shallow copy of this parameter.

Returns
-------
parameter:
    A shallow copy of this parameter.

◆ __deepcopy__()

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.

◆ update()

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.


The documentation for this class was generated from the following file: