Package org.nd4j.linalg.learning

Class AMSGradUpdater

java.lang.Object

org.nd4j.linalg.learning.AMSGradUpdater

All Implemented Interfaces:

GradientUpdater<AMSGrad>

public class AMSGradUpdater
extends Object
implements GradientUpdater<AMSGrad>

Field Summary

Fields

Modifier and Type	Field	Description
static String	M_STATE
static String	V_HAT_STATE
static String	V_STATE

Constructor Summary

Constructors

Constructor	Description
AMSGradUpdater(AMSGrad config)

Method Summary

Modifier and Type	Method	Description
void	applyUpdater(INDArray gradient, int iteration, int epoch)	Modify the gradient to be an update.
Map<String,INDArray>	getState()
void	setState(@NonNull Map<String,INDArray> stateMap, boolean initialize)
void	setStateViewArray(INDArray viewArray, long[] gradientShape, char gradientOrder, boolean initialize)	For the internal updater state (if any): set this to use the provided array.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.nd4j.linalg.learning.GradientUpdater

getConfig

Field Detail

M_STATE

public static final String M_STATE

See Also:

Constant Field Values

V_STATE

public static final String V_STATE

See Also:

Constant Field Values

V_HAT_STATE

public static final String V_HAT_STATE

See Also:

Constant Field Values

Constructor Detail

AMSGradUpdater

public AMSGradUpdater(AMSGrad config)

Method Detail

setState

public void setState(@NonNull
                     @NonNull Map<String,INDArray> stateMap,
                     boolean initialize)

Specified by:

setState in interface GradientUpdater<AMSGrad>

getState

public Map<String,INDArray> getState()

Specified by:

getState in interface GradientUpdater<AMSGrad>

setStateViewArray

public void setStateViewArray(INDArray viewArray,
                              long[] gradientShape,
                              char gradientOrder,
                              boolean initialize)

Description copied from interface: GradientUpdater

For the internal updater state (if any): set this to use the provided array. Used during initialization, and when restoring the updater state (after serialization, for example)

Specified by:

setStateViewArray in interface GradientUpdater<AMSGrad>

Parameters:

viewArray - Array (that is a view of a larger array) to use for the state.

initialize - If true: the updater must initialize the view array. If false: no change to view array contents

applyUpdater

public void applyUpdater(INDArray gradient,
                         int iteration,
                         int epoch)

Description copied from interface: GradientUpdater

Modify the gradient to be an update. Note that this is be done in-place

Specified by:

applyUpdater in interface GradientUpdater<AMSGrad>

Parameters:

gradient - the gradient to modify