diff --git a/src/mlpack/methods/ann/init_rules/network_init.hpp b/src/mlpack/methods/ann/init_rules/network_init.hpp index 4c923d6dc6..d0810b7840 100644 --- a/src/mlpack/methods/ann/init_rules/network_init.hpp +++ b/src/mlpack/methods/ann/init_rules/network_init.hpp @@ -71,7 +71,7 @@ class NetworkInitialization // Initialize the layer with the specified parameter/weight // initialization rule. const size_t weight = network[i]->WeightSize(); - arma::Mat tmp = arma::mat(parameters.memptr() + offset, + arma::Mat tmp = arma::Mat(parameters.memptr() + offset, weight, 1, false, false); initializeRule.Initialize(tmp, tmp.n_elem, 1);