Skip to content

Commit

Permalink
Change default tolerance to -1, meaning no tolerance-based stopping
Browse files Browse the repository at this point in the history
  • Loading branch information
danielhers committed Feb 8, 2017
1 parent 5d4e127 commit ce1c900
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions parser/lstm-parse.cc
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ void InitCommandLine(int argc, char** argv, po::variables_map* conf) {
("lstm_input_dim", po::value<unsigned>()->default_value(60), "LSTM input dimension")
("train,t", "Should training be run?")
("maxit,M", po::value<unsigned>()->default_value(8000), "Maximum number of training iterations")
("tolerance", po::value<double>()->default_value(0.0), "Tolerance on dev uas for stopping training")
("tolerance", po::value<double>()->default_value(-1.0), "Tolerance on dev uas for stopping training")
("words,w", po::value<string>(), "Pretrained word embeddings")
("use_spelling,S", "Use spelling model") //Miguel. Spelling model
("help,h", "Help");
Expand Down Expand Up @@ -967,7 +967,9 @@ int main(int argc, char** argv) {
const unsigned maxit = conf["maxit"].as<unsigned>();
cerr << "Maximum number of iterations: " << maxit << "\n";
const double tolerance = conf["tolerance"].as<double>();
cerr << "Optimization tolerance: " << tolerance << "\n";
if (tolerance > 0.0) {
cerr << "Optimization tolerance: " << tolerance << "\n";
}
ostringstream os;
os << "parser_" << (USE_POS ? "pos" : "nopos")
<< '_' << LAYERS
Expand Down Expand Up @@ -1060,7 +1062,7 @@ int main(int argc, char** argv) {
double uas = -1;
double prev_uas = -1;
while(!requested_stop && iter < maxit &&
(uas < 0 || prev_uas < 0 || abs(prev_uas - uas) > tolerance)) {
(tolerance < 0 || uas < 0 || prev_uas < 0 || abs(prev_uas - uas) > tolerance)) {
for (unsigned sii = 0; sii < status_every_i_iterations; ++sii) {
if (si == corpus.nsentences) {
si = 0;
Expand Down

0 comments on commit ce1c900

Please sign in to comment.