From 540ed79cc13b694264471d8cd8f4735e841707ae Mon Sep 17 00:00:00 2001
From: Chris Dyer <cdyer@cs.cmu.edu>
Date: Tue, 12 Jul 2011 23:32:11 -0400
Subject: minor optimization

---
 pro-train/mr_pro_reduce.cc | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'pro-train')

diff --git a/pro-train/mr_pro_reduce.cc b/pro-train/mr_pro_reduce.cc
index e1a7db8a..5382e1a5 100644
--- a/pro-train/mr_pro_reduce.cc
+++ b/pro-train/mr_pro_reduce.cc
@@ -149,18 +149,20 @@ int main(int argc, char** argv) {
 #endif
     cll += reg;
     cerr << cll << " (REG=" << reg << ")\tPPL=" << ppl << "\t";
-    bool failed = false;
     try {
-      opt.Optimize(cll, vg, &x);
+      vector<double> old_x = x;
+      do {
+        opt.Optimize(cll, vg, &x);
+        converged = opt.HasConverged();
+      } while (!converged && x == old_x);
     } catch (...) {
       cerr << "Exception caught, assuming convergence is close enough...\n";
-      failed = true;
+      converged = true;
     }
     if (fabs(x[0]) > MAX_BIAS) {
       cerr << "Biased model learned. Are your training instances wrong?\n";
       cerr << "  BIAS: " << x[0] << endl;
     }
-    converged = failed || opt.HasConverged();
   }
   Weights w;
   if (conf.count("weights")) {
-- 
cgit v1.2.3