[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