diff options
Diffstat (limited to 'training/dtrain')
-rw-r--r-- | training/dtrain/dtrain.cc | 2 | ||||
-rw-r--r-- | training/dtrain/dtrain.h | 6 |
2 files changed, 5 insertions, 3 deletions
diff --git a/training/dtrain/dtrain.cc b/training/dtrain/dtrain.cc index 42a771e0..b44f8f2b 100644 --- a/training/dtrain/dtrain.cc +++ b/training/dtrain/dtrain.cc @@ -68,12 +68,14 @@ main(int argc, char** argv) cerr << _p4; // output configuration cerr << "Parameters:" << endl; + cerr << setw(25) << "bitext " << "'" << input_fn << "'" << endl; cerr << setw(25) << "k " << k << endl; cerr << setw(25) << "score " << "'" << score_name << "'" << endl; cerr << setw(25) << "N " << N << endl; cerr << setw(25) << "T " << T << endl; cerr << setw(25) << "learning rate " << eta << endl; cerr << setw(25) << "margin " << margin << endl; + cerr << setw(25) << "average " << average << endl; cerr << setw(25) << "l1 reg " << l1_reg << endl; cerr << setw(25) << "decoder conf " << "'" << conf["decoder_conf"].as<string>() << "'" << endl; diff --git a/training/dtrain/dtrain.h b/training/dtrain/dtrain.h index 1d07defa..cdc2f628 100644 --- a/training/dtrain/dtrain.h +++ b/training/dtrain/dtrain.h @@ -65,12 +65,12 @@ dtrain_init(int argc, char** argv, po::variables_map* conf) "list of weights to print after each iteration"); po::options_description clopts("Command Line Options"); clopts.add_options() - ("conf,c", po::value<string>(), "dtrain configuration file") - ("help,h", po::bool_switch()->default_value(false), "display options"); + ("conf,c", po::value<string>(), "dtrain configuration file") + ("help,h", po::bool_switch(), "display options"); opts.add(clopts); po::store(parse_command_line(argc, argv, opts), *conf); cerr << "dtrain" << endl << endl; - if (conf->count("help")) { + if ((*conf)["help"].as<bool>()) { cerr << opts << endl; return false; |