diff options
author | Patrick Simianer <p@simianer.de> | 2015-06-09 14:39:42 +0200 |
---|---|---|
committer | Patrick Simianer <p@simianer.de> | 2015-06-09 14:39:42 +0200 |
commit | 5d1971d1eefaf19b363e8b12f16d6e778b910a5d (patch) | |
tree | 64c1bead3071715016e120e78b7c6bb0fe2d27f8 | |
parent | 629d84151f1288f8de11fc866c3960cfcf3ff0fd (diff) |
re-enable learning
-rw-r--r-- | training/dtrain/dtrain_net_interface.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/training/dtrain/dtrain_net_interface.cc b/training/dtrain/dtrain_net_interface.cc index d126ed27..265bc014 100644 --- a/training/dtrain/dtrain_net_interface.cc +++ b/training/dtrain/dtrain_net_interface.cc @@ -101,7 +101,7 @@ main(int argc, char** argv) } } } - + if (!next) break; @@ -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++; @@ -125,7 +125,7 @@ main(int argc, char** argv) string done = "done"; sock.send(done.c_str(), done.size()+1, 0); } // input loop - + if (output_fn != "") { cerr << "writing final weights to '" << output_fn << "'" << endl; lambdas.init_vector(decoder_weights); |