From 629d84151f1288f8de11fc866c3960cfcf3ff0fd Mon Sep 17 00:00:00 2001 From: Patrick Simianer Date: Wed, 27 May 2015 19:20:29 +0200 Subject: net: python extractor output --- training/dtrain/dtrain_net.h | 3 ++- training/dtrain/dtrain_net_interface.cc | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) (limited to 'training') 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()->default_value(100), "size of kbest list") ("N", po::value()->default_value(4), "N for BLEU approximation") ("margin,m", po::value()->default_value(0.), "margin for margin perceptron") - ("output,o", po::value()->default_value(""), "final weights file"); + ("output,o", po::value()->default_value(""), "final weights file") + ("input_weights,w", po::value(), "input weights file"); po::options_description cl("Command Line Options"); cl.add_options() ("conf,c", po::value(), "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++; -- cgit v1.2.3