org.deeplearning4j.arbiter.layers

Class ConvolutionLayerSpace.Builder

java.lang.Object

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

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

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

org.deeplearning4j.arbiter.layers.ConvolutionLayerSpace.Builder

Enclosing class:

ConvolutionLayerSpace

public static class ConvolutionLayerSpace.Builder
extends FeedForwardLayerSpace.Builder<ConvolutionLayerSpace.Builder>

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
ConvolutionLayerSpace	build()
ConvolutionLayerSpace.Builder	convolutionMode(org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)
ConvolutionLayerSpace.Builder	convolutionMode(ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode)
ConvolutionLayerSpace.Builder	dilation(int... dilation)
ConvolutionLayerSpace.Builder	dilation(ParameterSpace<int[]> dilation)
ConvolutionLayerSpace.Builder	hasBias(boolean hasBias)
ConvolutionLayerSpace.Builder	hasBias(ParameterSpace<Boolean> hasBias)
ConvolutionLayerSpace.Builder	kernelSize(int... kernelSize)
ConvolutionLayerSpace.Builder	kernelSize(ParameterSpace<int[]> kernelSize)
ConvolutionLayerSpace.Builder	padding(int... padding)
ConvolutionLayerSpace.Builder	padding(ParameterSpace<int[]> padding)
ConvolutionLayerSpace.Builder	stride(int... stride)
ConvolutionLayerSpace.Builder	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

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 ConvolutionLayerSpace.Builder dilation(int... dilation)

dilation

public ConvolutionLayerSpace.Builder dilation(ParameterSpace<int[]> dilation)

kernelSize

public ConvolutionLayerSpace.Builder kernelSize(int... kernelSize)

kernelSize

public ConvolutionLayerSpace.Builder kernelSize(ParameterSpace<int[]> kernelSize)

stride

public ConvolutionLayerSpace.Builder stride(int... stride)

stride

public ConvolutionLayerSpace.Builder stride(ParameterSpace<int[]> stride)

padding

public ConvolutionLayerSpace.Builder padding(int... padding)

padding

public ConvolutionLayerSpace.Builder padding(ParameterSpace<int[]> padding)

convolutionMode

public ConvolutionLayerSpace.Builder convolutionMode(org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

convolutionMode

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

hasBias

public ConvolutionLayerSpace.Builder hasBias(boolean hasBias)

hasBias

public ConvolutionLayerSpace.Builder hasBias(ParameterSpace<Boolean> hasBias)

build

public ConvolutionLayerSpace build()

Specified by:

build in class LayerSpace.Builder<ConvolutionLayerSpace.Builder>