[mlpack-git] master: Use the correct network output size. (96dc228)

gitdub at mlpack.org gitdub at mlpack.org
Tue Aug 30 16:50:21 EDT 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/459242594a29b6bae06ef3ffa4be7c2757c15246...96dc228a1cf7077e3ec00e420a62edec3b10daac

>---------------------------------------------------------------

commit 96dc228a1cf7077e3ec00e420a62edec3b10daac
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date:   Tue Aug 30 22:50:21 2016 +0200

    Use the correct network output size.


>---------------------------------------------------------------

96dc228a1cf7077e3ec00e420a62edec3b10daac
 src/mlpack/tests/convolutional_network_test.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/tests/convolutional_network_test.cpp b/src/mlpack/tests/convolutional_network_test.cpp
index e4e3a3e..34a9ee1 100644
--- a/src/mlpack/tests/convolutional_network_test.cpp
+++ b/src/mlpack/tests/convolutional_network_test.cpp
@@ -96,7 +96,7 @@ void BuildVanillaNetwork()
   BaseLayer2D<> baseLayer1;
   PoolingLayer<> poolingLayer1(2);
 
-  LinearMappingLayer<> linearLayer0(192, 10);
+  LinearMappingLayer<> linearLayer0(4608, 10);
   BiasLayer<> biasLayer2(10);
   SoftmaxLayer<> softmaxLayer0;
 




More information about the mlpack-git mailing list