public class Sgd extends Object implements IUpdater
Modifier and Type | Class and Description |
---|---|
static class |
Sgd.Builder |
Modifier and Type | Field and Description |
---|---|
static double |
DEFAULT_SGD_LR |
Constructor and Description |
---|
Sgd() |
Modifier and Type | Method and Description |
---|---|
void |
applySchedules(int iteration,
double newLearningRate)
Apply the new learning rate and any other schedules
|
Sgd |
clone()
Clone the updater
|
GradientUpdater |
instantiate(INDArray viewArray,
boolean initializeViewArray)
Create a new gradient updater
|
long |
stateSize(long numParams)
Determine the updater state size for the given number of parameters.
|
public static final double DEFAULT_SGD_LR
public long stateSize(long numParams)
IUpdater
public void applySchedules(int iteration, double newLearningRate)
IUpdater
applySchedules
in interface IUpdater
iteration
- Current iteration countnewLearningRate
- new learning rate to set for the updaterpublic GradientUpdater instantiate(INDArray viewArray, boolean initializeViewArray)
IUpdater
instantiate
in interface IUpdater
viewArray
- The updater state size view awayinitializeViewArray
- If true: initialise the updater stateCopyright © 2017. All rights reserved.