tesseract
5.0.0-alpha-619-ge9db
lstm_squashed_test.cc
Go to the documentation of this file.
1
// (C) Copyright 2017, Google Inc.
2
// Licensed under the Apache License, Version 2.0 (the "License");
3
// you may not use this file except in compliance with the License.
4
// You may obtain a copy of the License at
5
// http://www.apache.org/licenses/LICENSE-2.0
6
// Unless required by applicable law or agreed to in writing, software
7
// distributed under the License is distributed on an "AS IS" BASIS,
8
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
9
// See the License for the specific language governing permissions and
10
// limitations under the License.
11
12
#include "
lstm_test.h
"
13
14
namespace
tesseract
{
15
16
// Tests that a Squashed network learns correctly.
17
// Almost as fast as the 2d-lstm.
18
TEST_F
(
LSTMTrainerTest
, TestSquashed) {
19
// A 2-layer LSTM with a Squashed feature-extracting LSTM on the bottom, and
20
// a small convolution/maxpool below that.
21
// Match training conditions to those typically used with this spec:
22
// recoding on, adam on.
23
SetupTrainerEng(
"[1,32,0,1 Ct3,3,16 Mp3,3 Lfys48 Lbx96 O1c1]"
,
24
"SQU-2-layer-lstm"
,
/*recode*/
true
,
/*adam*/
true
);
25
double
lstm_2d_err = TrainIterations(
kTrainerIterations
* 3 / 2);
26
EXPECT_LT(lstm_2d_err, 80);
27
LOG
(
INFO
) <<
"********** < 80 ************\n"
;
28
TestIntMode(
kTrainerIterations
);
29
}
30
31
}
// namespace tesseract.
INFO
Definition:
log.h:29
tesseract::kTrainerIterations
const int kTrainerIterations
Definition:
lstm_test.h:35
tesseract::TEST_F
TEST_F(EquationFinderTest, IdentifySpecialText)
Definition:
equationdetect_test.cc:181
tesseract
Definition:
baseapi.h:65
lstm_test.h
tesseract::LSTMTrainerTest
Definition:
lstm_test.h:46
LOG
Definition:
cleanapi_test.cc:19
unittest
lstm_squashed_test.cc
Generated on Thu Jan 30 2020 14:22:21 for tesseract by
1.8.16