Merge pull request deeplearning4j#1492 from deeplearning4j/mw_bn_quirk

Fixed test error by setting nOut for bn.
vovietanh · May 2, 2016 · 8f5a611 · 8f5a611
2 parents bb723e8 + 14f3805
commit 8f5a611
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java b/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java
@@ -139,7 +139,7 @@ public void testBatchNorm() {
                 .list()
                 .layer(0, new DenseLayer.Builder().nIn(4).nOut(3).weightInit(WeightInit.XAVIER).activation("tanh").build())
                 .layer(1, new DenseLayer.Builder().nIn(3).nOut(2).weightInit(WeightInit.XAVIER).activation("tanh").build())
-                .layer(2, new BatchNormalization.Builder().build())
+                .layer(2, new BatchNormalization.Builder().nOut(2).build())
                 .layer(3, new org.deeplearning4j.nn.conf.layers.OutputLayer.Builder(LossFunctions.LossFunction.MCXENT)
                         .weightInit(WeightInit.XAVIER)
                         .activation("softmax")