.build()) .layer(2, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(128).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(5, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(256).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(10, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(512).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(15, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(512).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(20, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .layer(21, new DenseLayer.Builder().nOut(4096).build())
.build()) .layer(2, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(128).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(5, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(256).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(9, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(512).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(13, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build()) .padding(1, 1).nOut(512).cudnnAlgoMode(cudnnAlgoMode).build()) .layer(17, new SubsamplingLayer.Builder() .poolingType(SubsamplingLayer.PoolingType.MAX).kernelSize(2, 2) .stride(2, 2).build())
protected void setLayerOptionsBuilder(SubsamplingLayer.Builder builder, double[] values) { super.setLayerOptionsBuilder(builder, values); if (poolingType != null) builder.poolingType(poolingType.getValue(values)); if (kernelSize != null) builder.kernelSize(kernelSize.getValue(values)); if (stride != null) builder.stride(stride.getValue(values)); if (padding != null) builder.padding(padding.getValue(values)); }