Merge branch 'master' of git://github.com/redpony/cdec

1 files changed, 5 insertions, 4 deletions

diff --git a/decoder/decoder.cc b/decoder/decoder.cc
index 239c8620..95ff6270 100644
--- a/decoder/decoder.cc
+++ b/decoder/decoder.cc
@@ -763,10 +763,6 @@ bool DecoderImpl::Decode(const string& input, DecoderObserver* o) {
              PRWeightFunction<double, EdgeProb, double, ELengthWeightFunction> >(forest);
     cerr << "  Expected length  (words): " << res.r / res.p << "\t" << res << endl;
   }
-  if (conf.count("show_partition")) {
-    const prob_t z = Inside<prob_t, EdgeProb>(forest);
-    cerr << "  Init. partition     log(Z): " << log(z) << endl;
-  }
 
   for (int pass = 0; pass < rescoring_passes.size(); ++pass) {
     const RescoringPass& rp = rescoring_passes[pass];
@@ -793,6 +789,11 @@ bool DecoderImpl::Decode(const string& input, DecoderObserver* o) {
       if (!SILENT) forest_stats(forest,"  " + passtr +" forest",show_tree_structure,show_features,cur_weights,oracle.show_derivation);
     }
 
+    if (conf.count("show_partition")) {
+      const prob_t z = Inside<prob_t, EdgeProb>(forest);
+      cerr << "  " << passtr << " partition     log(Z): " << log(z) << endl;
+    }
+
     string fullbp = "beam_prune" + StringSuffixForRescoringPass(pass);
     string fulldp = "density_prune" + StringSuffixForRescoringPass(pass);
     maybe_prune(forest,conf,fullbp.c_str(),fulldp.c_str(),passtr,srclen);