org.deeplearning4j.arbiter.layers

Class BaseConvolutionLayerSpace.Builder<T>

java.lang.Object

org.deeplearning4j.arbiter.layers.LayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BaseConvolutionLayerSpace.Builder<T>

Direct Known Subclasses:

Deconvolution2DLayerSpace.Builder, SeparableConvolution2DLayerSpace.Builder

Enclosing class:

BaseConvolutionLayerSpace<T extends org.deeplearning4j.nn.conf.layers.FeedForwardLayer>

public abstract static class BaseConvolutionLayerSpace.Builder<T>
extends FeedForwardLayerSpace.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode>	convolutionMode
protected ParameterSpace<int[]>	dilation
protected ParameterSpace<Boolean>	hasBias
protected ParameterSpace<int[]>	kernelSize
protected ParameterSpace<int[]>	padding
protected ParameterSpace<int[]>	stride

Fields inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAll, constrainBias, constrainWeights, nIn, nOut

Fields inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

activationFunction, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, weightInit, weightNoise

Fields inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
T	convolutionMode(org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)
T	convolutionMode(ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode)
T	dilation(int... dilation)
T	dilation(ParameterSpace<int[]> dilation)
T	hasBias(boolean hasBias)
T	hasBias(ParameterSpace<Boolean> hasBias)
T	kernelSize(int... kernelSize)
T	kernelSize(ParameterSpace<int[]> kernelSize)
T	padding(int... padding)
T	padding(ParameterSpace<int[]> padding)
T	stride(int... stride)
T	stride(ParameterSpace<int[]> stride)

Methods inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAllParams, constrainAllParams, constrainBias, constrainBias, constrainWeights, constrainWeights, nIn, nIn, nOut, nOut

Methods inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

activation, activation, activation, activation, activationFn, biasInit, biasInit, biasUpdater, biasUpdater, dist, dist, gradientNormalization, gradientNormalization, gradientNormalizationThreshold, gradientNormalizationThreshold, l1, l1, l1Bias, l1Bias, l2, l2, l2Bias, l2Bias, updater, updater, weightInit, weightInit, weightInit

Methods inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

build, dropOut, dropOut, iDropOut

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

dilation

protected ParameterSpace<int[]> dilation

kernelSize

protected ParameterSpace<int[]> kernelSize

stride

protected ParameterSpace<int[]> stride

padding

protected ParameterSpace<int[]> padding

convolutionMode

protected ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode

hasBias

protected ParameterSpace<Boolean> hasBias

Constructor Detail

Builder

public Builder()

Method Detail

dilation

public T dilation(int... dilation)

dilation

public T dilation(ParameterSpace<int[]> dilation)

kernelSize

public T kernelSize(int... kernelSize)

kernelSize

public T kernelSize(ParameterSpace<int[]> kernelSize)

stride

public T stride(int... stride)

stride

public T stride(ParameterSpace<int[]> stride)

padding

public T padding(int... padding)

padding

public T padding(ParameterSpace<int[]> padding)

convolutionMode

public T convolutionMode(org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

convolutionMode

public T convolutionMode(ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode)

hasBias

public T hasBias(boolean hasBias)

hasBias

public T hasBias(ParameterSpace<Boolean> hasBias)