org.deeplearning4j.arbiter.conf.updater

Class AdaMaxSpace

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.AdaMaxSpace

All Implemented Interfaces:

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

public class AdaMaxSpace
extends BaseUpdaterSpace

Constructor Summary

Constructors

Constructor and Description
AdaMaxSpace(ParameterSpace<Double> learningRate)
AdaMaxSpace(ParameterSpace<Double> learningRate, ParameterSpace<Double> beta1, ParameterSpace<Double> beta2, ParameterSpace<Double> epsilon)
AdaMaxSpace(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 AdaMaxSpace	withLR(ParameterSpace<Double> lr)
static AdaMaxSpace	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

AdaMaxSpace

public AdaMaxSpace(ParameterSpace<Double> learningRate)

AdaMaxSpace

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

AdaMaxSpace

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

Method Detail

withLR

public static AdaMaxSpace withLR(ParameterSpace<Double> lr)

withLRSchedule

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

getValue

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