summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2015-06-09 14:39:42 +0200
committerPatrick Simianer <p@simianer.de>2015-06-09 14:39:42 +0200
commit5d1971d1eefaf19b363e8b12f16d6e778b910a5d (patch)
tree64c1bead3071715016e120e78b7c6bb0fe2d27f8
parent629d84151f1288f8de11fc866c3960cfcf3ff0fd (diff)
re-enable learning
-rw-r--r--training/dtrain/dtrain_net_interface.cc6
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);