Use ConstantStep from ensmallen.

This commit is contained in:
Marcus Edel
2018-11-15 22:32:18 +01:00
parent 0f35a28826
commit 4eec0b8f46
+2 -2
View File
@@ -343,11 +343,11 @@ BOOST_AUTO_TEST_CASE(BiasSVDFunctionParallelOptimize)
// Make the Bias SVD function and the optimizer.
BiasSVDFunction<arma::mat> biasSVDFunc(data, rank, lambda);
ConstantStep decayPolicy(alpha);
ens::ConstantStep decayPolicy(alpha);
// Iterate till convergence.
// The threadShareSize is chosen such that each function gets optimized.
ens::ParallelSGD<ConstantStep> optimizer(0,
ens::ParallelSGD<ens::ConstantStep> optimizer(0,
std::ceil((float) biasSVDFunc.NumFunctions() / omp_get_max_threads()), 1e-5,
true, decayPolicy);