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.BaseLayer.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 ConvolutionLayer.BwdDataAlgo	cudnnBwdDataAlgo
protected ConvolutionLayer.BwdFilterAlgo	cudnnBwdFilterAlgo
protected ConvolutionLayer.FwdAlgo	cudnnFwdAlgo
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.BaseLayer.Builder

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

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

dropOut, layerName

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.
T	cudnnBwdDataMode(ConvolutionLayer.BwdDataAlgo cudnnBwdDataAlgo)
T	cudnnBwdFilterMode(ConvolutionLayer.BwdFilterAlgo cudnnBwdFilterAlgo)
T	cudnnFwdMode(ConvolutionLayer.FwdAlgo cudnnFwdAlgo)

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

nIn, nOut

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

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

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

build, dropOut, name

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

cudnnFwdAlgo

protected ConvolutionLayer.FwdAlgo cudnnFwdAlgo

cudnnBwdFilterAlgo

protected ConvolutionLayer.BwdFilterAlgo cudnnBwdFilterAlgo

cudnnBwdDataAlgo

protected ConvolutionLayer.BwdDataAlgo cudnnBwdDataAlgo

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.

cudnnFwdMode

public T cudnnFwdMode(ConvolutionLayer.FwdAlgo cudnnFwdAlgo)

cudnnBwdFilterMode

public T cudnnBwdFilterMode(ConvolutionLayer.BwdFilterAlgo cudnnBwdFilterAlgo)

cudnnBwdDataMode

public T cudnnBwdDataMode(ConvolutionLayer.BwdDataAlgo cudnnBwdDataAlgo)