Minor change
This commit is contained in:
@@ -118,7 +118,8 @@ class GlorotInitialization
|
||||
/**
|
||||
* XavierInitilization is the popular name for this method.
|
||||
*/
|
||||
using XavierInitialization = GlorotInitialization<false>
|
||||
using XavierInitialization = GlorotInitialization<true>;
|
||||
using GlorotInitialization = GlorotInitialization<false>;
|
||||
} // namespace ann
|
||||
} // namespace mlpack
|
||||
|
||||
|
||||
@@ -294,7 +294,7 @@ BOOST_AUTO_TEST_CASE(GlorotInitUniformTest)
|
||||
arma::mat weights;
|
||||
arma::cube weights3d;
|
||||
|
||||
GlorotInitialization<true> glorotInit;
|
||||
XavierInitialization glorotInit;
|
||||
|
||||
glorotInit.Initialize(weights, 100, 100);
|
||||
glorotInit.Initialize(weights3d, 100, 100, 2);
|
||||
@@ -315,7 +315,7 @@ BOOST_AUTO_TEST_CASE(GlorotInitNormalTest)
|
||||
arma::mat weights;
|
||||
arma::cube weights3d;
|
||||
|
||||
GlorotInitialization<false> glorotInit;
|
||||
GlorotInitialization glorotInit;
|
||||
|
||||
glorotInit.Initialize(weights, 100, 100);
|
||||
glorotInit.Initialize(weights3d, 100, 100, 2);
|
||||
|
||||
Reference in New Issue
Block a user