diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/dtrain/dtrain_net.h | 3 | ||||
| -rw-r--r-- | training/dtrain/dtrain_net_interface.cc | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/training/dtrain/dtrain_net.h b/training/dtrain/dtrain_net.h index f6aa08b2..cbf171d6 100644 --- a/training/dtrain/dtrain_net.h +++ b/training/dtrain/dtrain_net.h @@ -43,7 +43,8 @@ dtrain_net_init(int argc, char** argv, po::variables_map* conf) ("k", po::value<size_t>()->default_value(100), "size of kbest list") ("N", po::value<size_t>()->default_value(4), "N for BLEU approximation") ("margin,m", po::value<weight_t>()->default_value(0.), "margin for margin perceptron") - ("output,o", po::value<string>()->default_value(""), "final weights file"); + ("output,o", po::value<string>()->default_value(""), "final weights file") + ("input_weights,w", po::value<string>(), "input weights file"); po::options_description cl("Command Line Options"); cl.add_options() ("conf,c", po::value<string>(), "dtrain configuration file") diff --git a/training/dtrain/dtrain_net_interface.cc b/training/dtrain/dtrain_net_interface.cc index 6c603040..d126ed27 100644 --- a/training/dtrain/dtrain_net_interface.cc +++ b/training/dtrain/dtrain_net_interface.cc @@ -117,7 +117,7 @@ main(int argc, char** argv) CollectUpdates(samples, updates, margin); cerr << "updates size " << updates.size() << endl; cerr << "lambdas before " << lambdas << endl; - lambdas.plus_eq_v_times_s(updates, 1.0); // FIXME: learning rate? + //lambdas.plus_eq_v_times_s(updates, 1.0); // FIXME: learning rate? cerr << "lambdas after " << lambdas << endl; i++; |