diff options
| author | Jean Senellart <jean@senellart.com> | 2017-12-21 10:17:26 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-21 10:17:26 +0300 |
| commit | aca8601ee208afc732d95f56b9d8f57d2f31b59d (patch) | |
| tree | d4feaad25eeb2c420accbd63cf8726102c2e1449 /onmt | |
| parent | eef498d14a53bf03e97908d441a3127a3c65e871 (diff) | |
fix beam size 1 broken with lexical_constraints (#470)
Diffstat (limited to 'onmt')
| -rw-r--r-- | onmt/translate/BeamSearcher.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/onmt/translate/BeamSearcher.lua b/onmt/translate/BeamSearcher.lua index ab03c734..e03feede 100644 --- a/onmt/translate/BeamSearcher.lua +++ b/onmt/translate/BeamSearcher.lua @@ -66,7 +66,7 @@ function BeamSearcher:search(beamSize, nBest, preFilterFactor, keepInitial) end local remaining = beams[1]:getRemaining() - if beams[1]:getTokens()[1]:size(1) ~= remaining * beamSize then + if beams[1]:getTokens()[1]:size(1) ~= remaining * self.beamSize then beams[1]:_replicate(self.beamSize) end local t = 1 |