| Package | Description |
|---|---|
| org.nd4j.linalg.learning.config | |
| org.nd4j.linalg.learning.regularization | |
| org.nd4j.linalg.schedule |
| Modifier and Type | Method and Description |
|---|---|
void |
AdaBelief.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
AdaDelta.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
AdaGrad.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
Adam.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
AdaMax.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
AMSGrad.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
IUpdater.setLrAndSchedule(double lr,
ISchedule lrSchedule)
Set the learning rate and schedule.
|
void |
Nadam.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
Nesterovs.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
NoOp.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
RmsProp.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
void |
Sgd.setLrAndSchedule(double lr,
ISchedule lrSchedule) |
| Constructor and Description |
|---|
AdaBelief(ISchedule learningRateSchedule) |
AdaGrad(ISchedule learningRateSchedule) |
AdaGrad(ISchedule learningRateSchedule,
double epsilon) |
Adam(ISchedule learningRateSchedule) |
AdaMax(ISchedule learningRateSchedule) |
AMSGrad(ISchedule learningRateSchedule) |
Nadam(ISchedule learningRateSchedule) |
Nesterovs(double learningRate,
ISchedule momentumSchedule) |
Nesterovs(ISchedule learningRateSchedule) |
Nesterovs(ISchedule learningRateSchedule,
double momentum) |
Nesterovs(ISchedule learningRateSchedule,
ISchedule momentumSchedule) |
RmsProp(ISchedule learningRateSchedule) |
Sgd(ISchedule learningRateSchedule) |
| Modifier and Type | Field and Description |
|---|---|
protected ISchedule |
WeightDecay.coeff |
protected ISchedule |
L1Regularization.l1 |
protected ISchedule |
L2Regularization.l2 |
| Constructor and Description |
|---|
L1Regularization(@NonNull ISchedule l1) |
L2Regularization(@NonNull ISchedule l2) |
WeightDecay(@NonNull ISchedule coeff,
boolean applyLR) |
| Modifier and Type | Class and Description |
|---|---|
class |
CycleSchedule |
class |
ExponentialSchedule |
class |
FixedSchedule |
class |
InverseSchedule |
class |
MapSchedule |
class |
PolySchedule |
class |
RampSchedule |
class |
SigmoidSchedule |
class |
StepSchedule |
| Modifier and Type | Field and Description |
|---|---|
protected ISchedule |
RampSchedule.baseSchedule |
| Modifier and Type | Method and Description |
|---|---|
ISchedule |
CycleSchedule.clone() |
ISchedule |
ExponentialSchedule.clone() |
ISchedule |
FixedSchedule.clone() |
ISchedule |
InverseSchedule.clone() |
ISchedule |
ISchedule.clone() |
ISchedule |
MapSchedule.clone() |
ISchedule |
RampSchedule.clone() |
ISchedule |
SigmoidSchedule.clone() |
ISchedule |
StepSchedule.clone() |
| Constructor and Description |
|---|
RampSchedule(ISchedule baseSchedule,
int numIter) |
Copyright © 2021. All rights reserved.