public abstract class BaseLayer extends Layer implements Serializable, Cloneable
Modifier and Type | Class and Description |
---|---|
static class |
BaseLayer.Builder<T extends BaseLayer.Builder<T>> |
Modifier and Type | Field and Description |
---|---|
protected IActivation |
activationFn |
protected double |
biasInit |
protected IUpdater |
biasUpdater |
protected Distribution |
dist |
protected GradientNormalization |
gradientNormalization |
protected double |
gradientNormalizationThreshold |
protected IUpdater |
iUpdater |
protected double |
l1 |
protected double |
l1Bias |
protected double |
l2 |
protected double |
l2Bias |
protected WeightInit |
weightInit |
protected IWeightNoise |
weightNoise |
constraints, iDropout, layerName
Constructor and Description |
---|
BaseLayer(BaseLayer.Builder builder) |
Modifier and Type | Method and Description |
---|---|
BaseLayer |
clone() |
GradientNormalization |
getGradientNormalization() |
IUpdater |
getUpdaterByParam(String paramName)
Get the updater for the given parameter.
|
void |
resetLayerDefaultConfig()
Reset the learning related configs of the layer to default.
|
getL1ByParam, getL2ByParam, getMemoryReport, getOutputType, getPreProcessorForInputType, initializeConstraints, initializer, instantiate, isPretrainParam, setNIn, setPretrain
equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getGradientNormalizationThreshold, getLayerName, isPretrain
protected IActivation activationFn
protected WeightInit weightInit
protected double biasInit
protected Distribution dist
protected double l1
protected double l2
protected double l1Bias
protected double l2Bias
protected IUpdater iUpdater
protected IUpdater biasUpdater
protected IWeightNoise weightNoise
protected GradientNormalization gradientNormalization
protected double gradientNormalizationThreshold
public BaseLayer(BaseLayer.Builder builder)
public void resetLayerDefaultConfig()
resetLayerDefaultConfig
in class Layer
public IUpdater getUpdaterByParam(String paramName)
getUpdaterByParam
in interface TrainingConfig
getUpdaterByParam
in class Layer
paramName
- Parameter namepublic GradientNormalization getGradientNormalization()
getGradientNormalization
in interface TrainingConfig
Copyright © 2018. All rights reserved.