public static class Convolution1DLayer.Builder extends ConvolutionLayer.BaseConvBuilder<Convolution1DLayer.Builder>
convolutionMode, cudnnAlgoMode, kernelSize, padding, stride
nIn, nOut
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 and Description |
---|
Builder() |
Builder(int kernelSize) |
Builder(int kernelSize,
int stride) |
Builder(int kernelSize,
int stride,
int padding) |
Modifier and Type | Method and Description |
---|---|
Convolution1DLayer |
build() |
Convolution1DLayer.Builder |
kernelSize(int kernelSize)
Size of the convolution
|
Convolution1DLayer.Builder |
padding(int padding) |
Convolution1DLayer.Builder |
stride(int stride) |
convolutionMode, cudnnAlgoMode
nIn, nOut
activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit
public Builder()
public Builder(int kernelSize, int stride)
public Builder(int kernelSize)
public Builder(int kernelSize, int stride, int padding)
public Convolution1DLayer.Builder kernelSize(int kernelSize)
kernelSize
- the length of the kernelpublic Convolution1DLayer.Builder stride(int stride)
public Convolution1DLayer.Builder padding(int padding)
public Convolution1DLayer build()
build
in class Layer.Builder<Convolution1DLayer.Builder>