Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Using @SuperBuilder for LayerConfigurations 

Signed-off-by: brian <brian@brutex.de>
master
Brian Rosenberger 2023-04-27 16:05:07 +02:00
parent cb236878a4
commit 35ea21e436
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cavis-dnn/cavis-dnn-nn/src/main/java/org/deeplearning4j/nn/conf/layers/Convolution1DLayer.java
View File

@ -183,9 +183,9 @@ public class Convolution1DLayer extends ConvolutionLayer {
return true; return true;
} }
private static final class Convolution1DLayerBuilderImpl extends ConvolutionLayerBuilder<ConvolutionLayer, Convolution1DLayerBuilderImpl> { private static final class Convolution1DLayerBuilderImpl extends Convolution1DLayerBuilder<Convolution1DLayer, Convolution1DLayerBuilderImpl> {
public ConvolutionLayer build() { public Convolution1DLayer build() {
ConvolutionLayer l = new ConvolutionLayer(this); Convolution1DLayer l = new Convolution1DLayer(this);
ConvolutionUtils.validateConvolutionModePadding(l.getConvolutionMode(), l.getPadding()); ConvolutionUtils.validateConvolutionModePadding(l.getConvolutionMode(), l.getPadding());
ConvolutionUtils.validateCnnKernelStridePadding( ConvolutionUtils.validateCnnKernelStridePadding(
l.getKernelSize(), l.getStride(), l.getPadding()); l.getKernelSize(), l.getStride(), l.getPadding());