org.deeplearning4j.arbiter.conf.updater

Class NesterovsSpace

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

All Implemented Interfaces:

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

public class NesterovsSpace
extends BaseUpdaterSpace

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<Double>	learningRate
protected ParameterSpace<org.nd4j.linalg.schedule.ISchedule>	learningRateSchedule
protected ParameterSpace<Double>	momentum
protected ParameterSpace<org.nd4j.linalg.schedule.ISchedule>	momentumSchedule

Constructor Summary

Constructors

Constructor and Description
NesterovsSpace(ParameterSpace<Double> learningRate)
NesterovsSpace(ParameterSpace<Double> learningRate, ParameterSpace<Double> momentum)
NesterovsSpace(ParameterSpace<Double> learningRate, ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule, ParameterSpace<Double> momentum, ParameterSpace<org.nd4j.linalg.schedule.ISchedule> momentumSchedule)

Method Summary

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

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

Field Detail

learningRate

protected ParameterSpace<Double> learningRate

learningRateSchedule

protected ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule

momentum

protected ParameterSpace<Double> momentum

momentumSchedule

protected ParameterSpace<org.nd4j.linalg.schedule.ISchedule> momentumSchedule

Constructor Detail

NesterovsSpace

public NesterovsSpace(ParameterSpace<Double> learningRate)

NesterovsSpace

public NesterovsSpace(ParameterSpace<Double> learningRate,
                      ParameterSpace<Double> momentum)

NesterovsSpace

public NesterovsSpace(ParameterSpace<Double> learningRate,
                      ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule,
                      ParameterSpace<Double> momentum,
                      ParameterSpace<org.nd4j.linalg.schedule.ISchedule> momentumSchedule)

Method Detail

withLR

public static NesterovsSpace withLR(ParameterSpace<Double> lr)

withLR

public static NesterovsSpace withLR(ParameterSpace<Double> lr,
                                    double momentum)

withLR

public static NesterovsSpace withLR(ParameterSpace<Double> lr,
                                    ParameterSpace<Double> momentum)

withLRSchedule

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

withLRSchedule

public static NesterovsSpace withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule,
                                            double momentum)

withLRSchedule

public static NesterovsSpace withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule,
                                            ParameterSpace<Double> momentum)

getValue

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