public interface Optimizer
Modifier and Type | Field and Description |
---|---|
static int |
COL_IDX |
static int |
DEPTH_IDX |
static float |
EPS
Smoothing term to prevent division by zero if sqr grad sum becomes zero 1e-8 should be also tried
https://d2l.ai/chapter_optimization/adagrad.html 1e-6
The value to use is 1e-6, 1e-8, Keras uses 1e-7 for adam
|
static int |
FOURTH_DIM_IDX |
static int |
ROW_IDX |
Modifier and Type | Method and Description |
---|---|
float |
calculateDeltaBias(float gradient,
int idx) |
float |
calculateDeltaWeight(float gradient,
int... index) |
static Optimizer |
create(OptimizerType type,
AbstractLayer layer)
Factory method to create different types of optimizers
|
void |
setLearningRate(float learningRate) |
static final float EPS
static final int ROW_IDX
static final int COL_IDX
static final int DEPTH_IDX
static final int FOURTH_DIM_IDX
float calculateDeltaWeight(float gradient, int... index)
float calculateDeltaBias(float gradient, int idx)
void setLearningRate(float learningRate)
static Optimizer create(OptimizerType type, AbstractLayer layer)
type
- layer
- Copyright © 2022. All rights reserved.