marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-11-30 21:39:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixed scoreBreakdown

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2016-04-20 01:06:40 +02:00
parent 712e50b2d5
commit 1f6049a04e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/decoder/search.h
View File

@ -142,8 +142,11 @@ class Search {
hyp->GetCostBreakdown()[0] = breakDowns[0][i];
else {
float cost = 0;
if(j < ProbsEnsemble.size())
if(j < ProbsEnsemble.size()) {
if(prevHyps[hypIndex]->GetCostBreakdown().size() < ProbsEnsemble.size())
const_cast<HypothesisPtr&>(prevHyps[hypIndex])->GetCostBreakdown().resize(ProbsEnsemble.size(), 0.0);
cost = breakDowns[j][i] + const_cast<HypothesisPtr&>(prevHyps[hypIndex])->GetCostBreakdown()[j];
}
sum += weights[j] * cost;
hyp->GetCostBreakdown()[j] = cost;
}