From 71bcde9be6a955d4096ad5ae1f2feb4cfa604bde Mon Sep 17 00:00:00 2001 From: Marcus Edel Date: Tue, 7 Nov 2017 00:49:28 +0100 Subject: [PATCH] Use the correct batch size for the LSTM/FastLSTM layer test. --- src/mlpack/tests/ann_layer_test.cpp | 4 ++-- src/mlpack/tests/recurrent_network_test.cpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp index 6456902ab2..9c803a3416 100644 --- a/src/mlpack/tests/ann_layer_test.cpp +++ b/src/mlpack/tests/ann_layer_test.cpp @@ -766,7 +766,7 @@ BOOST_AUTO_TEST_CASE(LSTMRrhoTest) modelB.Add >(10, 3); modelB.Add >(); - 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 >(10, 3); modelB.Add >(); - 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); diff --git a/src/mlpack/tests/recurrent_network_test.cpp b/src/mlpack/tests/recurrent_network_test.cpp index 5fe9182b57..38857ef9ef 100644 --- a/src/mlpack/tests/recurrent_network_test.cpp +++ b/src/mlpack/tests/recurrent_network_test.cpp @@ -455,7 +455,7 @@ void ReberGrammarTestNetwork(const size_t hiddenSize = 4, model.Add >(hiddenSize, outputSize); model.Add >(); MomentumSGD opt(0.06, 50, 2, -50000); - + arma::mat inputTemp, labelsTemp; for (size_t i = 0; i < (iterations + offset); i++) {