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

Use the correct batch size for the LSTM/FastLSTM layer test.

Browse Source
This commit is contained in:
Marcus Edel
2017-11-07 00:49:28 +01:00
parent 43c1017278
commit 71bcde9be6
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/ann_layer_test.cpp
+2 -2
View File
@@ -766,7 +766,7 @@ BOOST_AUTO_TEST_CASE(LSTMRrhoTest)
modelB.Add<LSTM<> >(10, 3);
modelB.Add<LogSoftMax<> >();
optimization::StandardSGD opt(0.1, 5, -100, false);
optimization::StandardSGD opt(0.1, 1, 5, -100, false);
modelA.Train(input, target, opt);
modelB.Train(input, target, opt);
@@ -846,7 +846,7 @@ BOOST_AUTO_TEST_CASE(FastLSTMRrhoTest)
modelB.Add<FastLSTM<> >(10, 3);
modelB.Add<LogSoftMax<> >();
optimization::StandardSGD opt(0.1, 5, -100, false);
optimization::StandardSGD opt(0.1, 1, 5, -100, false);
modelA.Train(input, target, opt);
modelB.Train(input, target, opt);
src/mlpack/tests/recurrent_network_test.cpp
+1 -1
View File
@@ -455,7 +455,7 @@ void ReberGrammarTestNetwork(const size_t hiddenSize = 4,
model.Add<Linear<> >(hiddenSize, outputSize);
model.Add<SigmoidLayer<> >();
MomentumSGD opt(0.06, 50, 2, -50000);
arma::mat inputTemp, labelsTemp;
for (size_t i = 0; i < (iterations + offset); i++)
{