From 8f08fd440c7df10288ead41f9552d2e9cb06f1ae Mon Sep 17 00:00:00 2001 From: Paul Dubs Date: Thu, 29 Mar 2018 14:08:42 +0200 Subject: [PATCH] Fix Copy&Paste Error in VariationalAutoencoder Builder --- .../nn/conf/layers/variational/VariationalAutoencoder.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/layers/variational/VariationalAutoencoder.java b/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/layers/variational/VariationalAutoencoder.java index 05cc5b5f3036..b5d070944e26 100644 --- a/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/layers/variational/VariationalAutoencoder.java +++ b/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/layers/variational/VariationalAutoencoder.java @@ -185,7 +185,7 @@ public Builder encoderLayerSizes(int... encoderLayerSizes) { * @param decoderLayerSizes Size of each deccoder layer in the variational autoencoder */ public Builder decoderLayerSizes(int... decoderLayerSizes) { - if (encoderLayerSizes == null || encoderLayerSizes.length < 1) { + if (decoderLayerSizes == null || decoderLayerSizes.length < 1) { throw new IllegalArgumentException("Decoder layer sizes array must have length > 0"); } this.decoderLayerSizes = decoderLayerSizes;