org.deeplearning4j.arbiter.conf.updater

Class AdamSpace

java.lang.Object

org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace<org.nd4j.linalg.learning.config.IUpdater>

org.deeplearning4j.arbiter.conf.updater.BaseUpdaterSpace

org.deeplearning4j.arbiter.conf.updater.AdamSpace

All Implemented Interfaces:

ParameterSpace<org.nd4j.linalg.learning.config.IUpdater>

public class AdamSpace
extends BaseUpdaterSpace

Constructor Summary

Constructors

Modifier	Constructor and Description
	AdamSpace(ParameterSpace<Double> learningRate)
	AdamSpace(ParameterSpace<Double> learningRate, ParameterSpace<Double> beta1, ParameterSpace<Double> beta2, ParameterSpace<Double> epsilon)
protected	AdamSpace(ParameterSpace<Double> learningRate, ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule, ParameterSpace<Double> beta1, ParameterSpace<Double> beta2, ParameterSpace<Double> epsilon)

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.learning.config.IUpdater	getValue(double[] parameterValues)
static AdamSpace	withLR(ParameterSpace<Double> lr)
static AdamSpace	withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

Methods inherited from class org.deeplearning4j.arbiter.conf.updater.BaseUpdaterSpace

collectLeaves, isLeaf, numParameters, setIndices

Methods inherited from class org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace

getNestedSpaces

Methods inherited from class java.lang.Object

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

Constructor Detail

AdamSpace

public AdamSpace(ParameterSpace<Double> learningRate)

AdamSpace

public AdamSpace(ParameterSpace<Double> learningRate,
                 ParameterSpace<Double> beta1,
                 ParameterSpace<Double> beta2,
                 ParameterSpace<Double> epsilon)

AdamSpace

protected AdamSpace(ParameterSpace<Double> learningRate,
                    ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule,
                    ParameterSpace<Double> beta1,
                    ParameterSpace<Double> beta2,
                    ParameterSpace<Double> epsilon)

Method Detail

withLR

public static AdamSpace withLR(ParameterSpace<Double> lr)

withLRSchedule

public static AdamSpace withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

getValue

public org.nd4j.linalg.learning.config.IUpdater getValue(double[] parameterValues)