summaryrefslogtreecommitdiff
path: root/pro-train
diff options
context:
space:
mode:
authorChris Dyer <cdyer@cab.ark.cs.cmu.edu>2012-05-26 20:59:00 -0400
committerChris Dyer <cdyer@cab.ark.cs.cmu.edu>2012-05-26 20:59:00 -0400
commit01ecc09f8e3a82c32bf7dd2f90c12554becea71d (patch)
tree205120623ecec74095d3d0fbaac22f35105febe4 /pro-train
parent427d2be0040c28bba680c48aca9514ef932ecb56 (diff)
fix incorrect interface use
Diffstat (limited to 'pro-train')
-rw-r--r--pro-train/mr_pro_reduce.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/pro-train/mr_pro_reduce.cc b/pro-train/mr_pro_reduce.cc
index 9698bb5d..5ef9b470 100644
--- a/pro-train/mr_pro_reduce.cc
+++ b/pro-train/mr_pro_reduce.cc
@@ -186,8 +186,9 @@ double LearnParameters(const vector<pair<bool, SparseVector<weight_t> > >& train
const unsigned memory_buffers,
const vector<weight_t>& prev_x,
vector<weight_t>* px) {
+ assert(px->size() == prev_x.size());
ProLoss loss(training, testing, C, T, prev_x);
- LBFGS<ProLoss> lbfgs(px, loss, C1, memory_buffers);
+ LBFGS<ProLoss> lbfgs(px, loss, memory_buffers, C1);
lbfgs.MinimizeFunction();
return loss.tppl;
}