org.deeplearning4j.arbiter.layers

Class SubsamplingLayerSpace.Builder

java.lang.Object

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

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

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

org.deeplearning4j.arbiter.layers.SubsamplingLayerSpace.Builder

Enclosing class:

SubsamplingLayerSpace

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

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode>	convolutionMode
protected ParameterSpace<int[]>	dilation
protected ParameterSpace<Double>	eps
protected ParameterSpace<int[]>	kernelSize
protected ParameterSpace<int[]>	padding
protected ParameterSpace<Integer>	pnorm
protected ParameterSpace<org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType>	poolingType
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
SubsamplingLayerSpace	build()
SubsamplingLayerSpace.Builder	convolutionMode(org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)
SubsamplingLayerSpace.Builder	convolutionMode(ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode)
SubsamplingLayerSpace.Builder	dilation(int... dilation)
SubsamplingLayerSpace.Builder	dilation(ParameterSpace<int[]> dilation)
SubsamplingLayerSpace.Builder	eps(double eps)
SubsamplingLayerSpace.Builder	eps(ParameterSpace<Double> eps)
SubsamplingLayerSpace.Builder	kernelSize(int... kernelSize)
SubsamplingLayerSpace.Builder	kernelSize(ParameterSpace<int[]> kernelSize)
SubsamplingLayerSpace.Builder	padding(int... padding)
SubsamplingLayerSpace.Builder	padding(ParameterSpace<int[]> padding)
SubsamplingLayerSpace.Builder	pnorm(int pnorm)
SubsamplingLayerSpace.Builder	pnorm(ParameterSpace<Integer> pnorm)
SubsamplingLayerSpace.Builder	poolingType(ParameterSpace<org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType> poolingType)
SubsamplingLayerSpace.Builder	poolingType(org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType poolingType)
SubsamplingLayerSpace.Builder	stride(int... stride)
SubsamplingLayerSpace.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

convolutionMode

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

poolingType

protected ParameterSpace<org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType> poolingType

dilation

protected ParameterSpace<int[]> dilation

kernelSize

protected ParameterSpace<int[]> kernelSize

stride

protected ParameterSpace<int[]> stride

padding

protected ParameterSpace<int[]> padding

pnorm

protected ParameterSpace<Integer> pnorm

eps

protected ParameterSpace<Double> eps

Constructor Detail

Builder

public Builder()

Method Detail

convolutionMode

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

convolutionMode

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

poolingType

public SubsamplingLayerSpace.Builder poolingType(org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType poolingType)

poolingType

public SubsamplingLayerSpace.Builder poolingType(ParameterSpace<org.deeplearning4j.nn.conf.layers.SubsamplingLayer.PoolingType> poolingType)

dilation

public SubsamplingLayerSpace.Builder dilation(int... dilation)

dilation

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

kernelSize

public SubsamplingLayerSpace.Builder kernelSize(int... kernelSize)

kernelSize

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

stride

public SubsamplingLayerSpace.Builder stride(int... stride)

stride

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

padding

public SubsamplingLayerSpace.Builder padding(int... padding)

padding

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

pnorm

public SubsamplingLayerSpace.Builder pnorm(int pnorm)

pnorm

public SubsamplingLayerSpace.Builder pnorm(ParameterSpace<Integer> pnorm)

eps

public SubsamplingLayerSpace.Builder eps(double eps)

eps

public SubsamplingLayerSpace.Builder eps(ParameterSpace<Double> eps)

build

public SubsamplingLayerSpace build()

Specified by:

build in class LayerSpace.Builder<SubsamplingLayerSpace.Builder>