apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor change

Browse Source
This commit is contained in:
Prabhat
2018-03-13 08:53:37 +05:30
parent 4c02d00cef
commit 330c838575
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/init_rules/glorot_init.hpp
+2 -2
View File
@@ -96,7 +96,7 @@ inline void GlorotInitializationType<false>::Initialize(arma::Mat<eT>& W,
if (W.is_empty())
W = arma::mat(rows, cols);
double_t var = (double)(2)/(rows + cols);
double var = 2.0/double(rows + cols);
GaussianInitialization normalInit(0.0, var);
normalInit.Initialize(W, rows, cols);
}
@@ -110,7 +110,7 @@ inline void GlorotInitializationType<true>::Initialize(arma::Mat<eT>& W,
if (W.is_empty())
W = arma::mat(rows, cols);
double_t a = sqrt(6) / sqrt(rows + cols); // limit of distribution
double a = sqrt(6) / sqrt(rows + cols); // limit of distribution
RandomInitialization randomInit(-a, a);
randomInit.Initialize(W, rows, cols);
}