summaryrefslogtreecommitdiff
path: root/training/mpi_online_optimize.cc
diff options
context:
space:
mode:
Diffstat (limited to 'training/mpi_online_optimize.cc')
-rw-r--r--training/mpi_online_optimize.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/training/mpi_online_optimize.cc b/training/mpi_online_optimize.cc
index 6f5988a4..0c032c01 100644
--- a/training/mpi_online_optimize.cc
+++ b/training/mpi_online_optimize.cc
@@ -277,6 +277,7 @@ int main(int argc, char** argv) {
rng = new MT19937;
}
SparseVector<double> x;
+ weights.InitSparseVector(&x);
int miter = corpus.size(); // hack to cause initial broadcast of order info
TrainingObserver observer;
double objective = 0;
@@ -308,7 +309,7 @@ int main(int argc, char** argv) {
broadcast(world, order, 0);
}
if (rank == 0)
- cerr << "Starting decoding. minibatch=" << size_per_proc << " sentences/proc x " << size << " procs. num_feats=" << x.size() << " training data proc. = " << (iter * batch_size / static_cast<double>(corpus.size())) << " eta=" << lr->eta(iter) << endl;
+ cerr << "iter=" << iter << " minibatch=" << size_per_proc << " sentences/proc x " << size << " procs. num_feats=" << x.size() << " passes_thru_data=" << (iter * batch_size / static_cast<double>(corpus.size())) << " eta=" << lr->eta(iter) << endl;
const int beg = size * miter * size_per_proc + rank * size_per_proc;
const int end = beg + size_per_proc;