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

Fix glorot variable name

Browse Source 
Signed-off-by: Omar Shrit <omar@avontech.fr>
This commit is contained in:
Omar Shrit
2024-03-07 14:00:37 +01:00
parent 5ad51c9ea8
commit 816f7b2575
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/init_rules/glorot_init.hpp
+4 -4
View File
@@ -122,8 +122,8 @@ inline void GlorotInitializationType<false>::Initialize(MatType& W,
if (W.is_empty())
W.set_size(rows, cols);
double var = 2.0 / double(rows + cols);
GaussianInitialization normalInit(0.0, var);
double stddev = 2.0 / double(rows + cols);
GaussianInitialization normalInit(0.0, stddev);
normalInit.Initialize(W, rows, cols);
}
@@ -150,8 +150,8 @@ inline void GlorotInitializationType<false>::Initialize(MatType& W,
if (W.is_empty())
Log::Fatal << "Cannot initialize an empty matrix." << std::endl;
double var = 2.0 / double(W.n_rows + W.n_cols);
GaussianInitialization normalInit(0.0, var);
double stddev = 2.0 / double(W.n_rows + W.n_cols);
GaussianInitialization normalInit(0.0, stddev);
normalInit.Initialize(W);
}