[mlpack-git] master: Use the naive convolution method as default. (2598e5a)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed Jul 1 16:29:39 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/cf08dc96724f70468c8e31ced1761ed000b55b18...222a7e191f8a7925a4870ce1acbd68589899dfde
>---------------------------------------------------------------
commit 2598e5aac2fe275951e25d874bd2638d2744c8d8
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Tue Jun 30 17:09:36 2015 +0200
Use the naive convolution method as default.
>---------------------------------------------------------------
2598e5aac2fe275951e25d874bd2638d2744c8d8
src/mlpack/methods/ann/connections/conv_connection.hpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/mlpack/methods/ann/connections/conv_connection.hpp b/src/mlpack/methods/ann/connections/conv_connection.hpp
index 748e219..26ebd10 100644
--- a/src/mlpack/methods/ann/connections/conv_connection.hpp
+++ b/src/mlpack/methods/ann/connections/conv_connection.hpp
@@ -10,7 +10,7 @@
#define __MLPACK_METHODS_ANN_CONNECTIONS_CONV_CONNECTION_HPP
#include <mlpack/core.hpp>
-#include <mlpack/methods/ann/init_rules/random_init.hpp>
+#include <mlpack/methods/ann/init_rules/nguyen_widrow_init.hpp>
#include <mlpack/methods/ann/optimizer/steepest_descent.hpp>
#include <mlpack/methods/ann/convolution_rules/border_modes.hpp>
#include <mlpack/methods/ann/convolution_rules/naive_convolution.hpp>
@@ -45,10 +45,10 @@ template<
typename InputLayerType,
typename OutputLayerType,
template<typename, typename> class OptimizerType = mlpack::ann::RMSPROP,
- class WeightInitRule = RandomInitialization,
+ class WeightInitRule = NguyenWidrowInitialization,
typename ForwardConvolutionRule = NaiveConvolution<ValidConvolution>,
- typename BackwardConvolutionRule = FFTConvolution<FullConvolution>,
- typename GradientConvolutionRule = FFTConvolution<ValidConvolution>,
+ typename BackwardConvolutionRule = NaiveConvolution<FullConvolution>,
+ typename GradientConvolutionRule = NaiveConvolution<ValidConvolution>,
typename DataType = arma::cube
>
class ConvConnection
More information about the mlpack-git
mailing list