[mlpack-git] master: Use the correct cnn network structure for the convolution network test. (8146057)
gitdub at mlpack.org
gitdub at mlpack.org
Tue Aug 30 11:48:24 EDT 2016
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/a8a8a1381b529a01420de6e792a4a1e7bd58a626...814605744b0a7c8b70a955076d3f110e2d6687db
>---------------------------------------------------------------
commit 814605744b0a7c8b70a955076d3f110e2d6687db
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Tue Aug 30 17:48:24 2016 +0200
Use the correct cnn network structure for the convolution network test.
>---------------------------------------------------------------
814605744b0a7c8b70a955076d3f110e2d6687db
src/mlpack/tests/convolutional_network_test.cpp | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/src/mlpack/tests/convolutional_network_test.cpp b/src/mlpack/tests/convolutional_network_test.cpp
index 1ca68c2..e4e3a3e 100644
--- a/src/mlpack/tests/convolutional_network_test.cpp
+++ b/src/mlpack/tests/convolutional_network_test.cpp
@@ -102,9 +102,7 @@ void BuildVanillaNetwork()
OneHotLayer outputLayer;
- auto modules = std::tie(convLayer0, biasLayer0, baseLayer0, poolingLayer0,
- convLayer1, biasLayer1, baseLayer1, poolingLayer1,
- linearLayer0, biasLayer2, softmaxLayer0);
+ auto modules = std::tie(convLayer0, baseLayer0, linearLayer0, softmaxLayer0);
CNN<decltype(modules), decltype(outputLayer),
RandomInitialization, MeanSquaredErrorFunction> net(modules, outputLayer);
More information about the mlpack-git
mailing list