[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