From 1b0d40959f529b67db3b9d10dbf93101e0c65c7c Mon Sep 17 00:00:00 2001
From: Patrick Simianer
Date: Mon, 27 Jan 2014 10:40:14 +0100
Subject: verbose parameter for mira (thanks Felix!)
---
training/mira/kbest_cut_mira.cc | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
(limited to 'training/mira/kbest_cut_mira.cc')
diff --git a/training/mira/kbest_cut_mira.cc b/training/mira/kbest_cut_mira.cc
index 990609d7..9415909e 100644
--- a/training/mira/kbest_cut_mira.cc
+++ b/training/mira/kbest_cut_mira.cc
@@ -95,7 +95,8 @@ bool InitCommandLine(int argc, char** argv, po::variables_map* conf) {
("stream,t", "Stream mode (used for realtime)")
("weights_output,O",po::value(),"Directory to write weights to")
("output_dir,D",po::value(),"Directory to place output in")
- ("decoder_config,c",po::value(),"Decoder configuration file");
+ ("decoder_config,c",po::value(),"Decoder configuration file")
+ ("verbose,v",po::value()->zero_tokens(),"verbose stderr output");
po::options_description clo("Command line options");
clo.add_options()
("config", po::value(), "Configuration file")
@@ -627,6 +628,7 @@ int main(int argc, char** argv) {
vector corpus;
+ const bool VERBOSE = conf.count("verbose");
const string metric_name = conf["mt_metric"].as();
optimizer = conf["optimizer"].as();
fear_select = conf["fear"].as();
@@ -790,7 +792,8 @@ int main(int argc, char** argv) {
double margin = cur_bad.features.dot(dense_weights) - cur_good.features.dot(dense_weights);
double mt_loss = (cur_good.mt_metric - cur_bad.mt_metric);
const double loss = margin + mt_loss;
- cerr << "LOSS: " << loss << " Margin:" << margin << " BLEUL:" << mt_loss << " " << cur_bad.features.dot(dense_weights) << " " << cur_good.features.dot(dense_weights) < 0.0 || !checkloss) {
SparseVector diff = cur_good.features;
diff -= cur_bad.features;
@@ -928,7 +931,7 @@ int main(int argc, char** argv) {
lambdas += (cur_pair[1]->features) * step_size;
lambdas -= (cur_pair[0]->features) * step_size;
- cerr << " Lambdas " << lambdas << endl;
+ if (VERBOSE) cerr << " Lambdas " << lambdas << endl;
//reload weights based on update
dense_weights.clear();
--
cgit v1.2.3