org.deeplearning4j.nn.conf.layers

Class ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder<T>

Direct Known Subclasses:

Convolution1DLayer.Builder, ConvolutionLayer.Builder

Enclosing class:

ConvolutionLayer

protected abstract static class ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>
extends FeedForwardLayer.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected ConvolutionMode	convolutionMode
protected ConvolutionLayer.AlgoMode	cudnnAlgoMode
protected int[]	kernelSize
protected int[]	padding
protected int[]	stride

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, nOut

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseConvBuilder()
protected	BaseConvBuilder(int... kernelSize)
protected	BaseConvBuilder(int[] kernelSize, int[] stride)
protected	BaseConvBuilder(int[] kernelSize, int[] stride, int[] padding)

Method Summary

Modifier and Type	Method and Description
T	convolutionMode(ConvolutionMode convolutionMode) Set the convolution mode for the Convolution layer.
T	cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode) Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, nOut

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, build, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit

Methods inherited from class java.lang.Object

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

Field Detail

convolutionMode

protected ConvolutionMode convolutionMode

kernelSize

protected int[] kernelSize

stride

protected int[] stride

padding

protected int[] padding

cudnnAlgoMode

protected ConvolutionLayer.AlgoMode cudnnAlgoMode

Constructor Detail

BaseConvBuilder

protected BaseConvBuilder(int[] kernelSize,
                          int[] stride,
                          int[] padding)

BaseConvBuilder

protected BaseConvBuilder(int[] kernelSize,
                          int[] stride)

BaseConvBuilder

protected BaseConvBuilder(int... kernelSize)

BaseConvBuilder

protected BaseConvBuilder()

Method Detail

convolutionMode

public T convolutionMode(ConvolutionMode convolutionMode)

Set the convolution mode for the Convolution layer. See ConvolutionMode for more details

Parameters:

convolutionMode - Convolution mode for layer

cudnnAlgoMode

public T cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)

Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.