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
commite331ea8e69489cfd727c0ad106c76efa69f3e06c (patch)
tree286abadcc9cbf73fd4502d171f72c3ea9d14200e /pro-train
parentbf0278b9492d678d8d7f601ea1513b896f2ad105 (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;
}