org.deeplearning4j.arbiter.conf.updater

Class SgdSpace

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

All Implemented Interfaces:

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

public class SgdSpace
extends BaseUpdaterSpace

Field Summary

Fields

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.learning.config.IUpdater	getValue(double[] parameterValues)

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

Constructor Detail

SgdSpace

public SgdSpace(ParameterSpace<Double> learningRate)

SgdSpace

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

Method Detail

getValue

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