org.deeplearning4j.nn.conf.layers

Class ConvolutionLayer.Builder

java.lang.Object

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

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

org.deeplearning4j.nn.conf.layers.ConvolutionLayer.Builder

Enclosing class:

ConvolutionLayer

public static class ConvolutionLayer.Builder
extends FeedForwardLayer.Builder<ConvolutionLayer.Builder>

Field Summary

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

nIn, nOut

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

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

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(int... kernelSize)
Builder(int[] kernelSize, int[] stride)
Builder(int[] kernelSize, int[] stride, int[] padding)

Method Summary

Modifier and Type	Method and Description
ConvolutionLayer	build()
ConvolutionLayer.Builder	convolutionMode(ConvolutionMode convolutionMode) Set the convolution mode for the Convolution layer.
ConvolutionLayer.Builder	convolutionType(org.nd4j.linalg.convolution.Convolution.Type convolutionType) Deprecated. Use convolutionMode
ConvolutionLayer.Builder	cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode) Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
ConvolutionLayer.Builder	kernelSize(int... kernelSize) Size of the convolution rows/columns
ConvolutionLayer.Builder	padding(int... padding)
ConvolutionLayer.Builder	stride(int... stride)

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, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l2, 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

Constructor Detail

Builder

public Builder(int[] kernelSize,
               int[] stride,
               int[] padding)

Builder

public Builder(int[] kernelSize,
               int[] stride)

Builder

public Builder(int... kernelSize)

Builder

public Builder()

Method Detail

convolutionType

@Deprecated
public ConvolutionLayer.Builder convolutionType(org.nd4j.linalg.convolution.Convolution.Type convolutionType)

Deprecated. Use convolutionMode

convolutionMode

public ConvolutionLayer.Builder convolutionMode(ConvolutionMode convolutionMode)

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

Parameters:

convolutionMode - Convolution mode for layer

kernelSize

public ConvolutionLayer.Builder kernelSize(int... kernelSize)

Size of the convolution rows/columns

Parameters:

kernelSize - the height and width of the kernel

Returns:

stride

public ConvolutionLayer.Builder stride(int... stride)

padding

public ConvolutionLayer.Builder padding(int... padding)

cudnnAlgoMode

public ConvolutionLayer.Builder cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)

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

build

public ConvolutionLayer build()

Specified by:

build in class Layer.Builder<ConvolutionLayer.Builder>