diff options
| author | Kenneth Heafield <github@kheafield.com> | 2012-10-10 19:02:40 +0100 |
|---|---|---|
| committer | Kenneth Heafield <github@kheafield.com> | 2012-10-10 19:02:40 +0100 |
| commit | afd7da3b2338661657ad0c4e9eec681e014d37bf (patch) | |
| tree | 6e1a274389f0b8a7931f14a33d5ef778e86d90ec /klm/lm/search_trie.hh | |
| parent | f7175481a7ea9b8733bc023016112141e29fd827 (diff) | |
KenLM 8307220
Diffstat (limited to 'klm/lm/search_trie.hh')
| -rw-r--r-- | klm/lm/search_trie.hh | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/klm/lm/search_trie.hh b/klm/lm/search_trie.hh index 10b22ab1..1264baf5 100644 --- a/klm/lm/search_trie.hh +++ b/klm/lm/search_trie.hh @@ -44,8 +44,8 @@ template <class Quant, class Bhiksha> class TrieSearch { Bhiksha::UpdateConfigFromBinary(fd, config); } - static std::size_t Size(const std::vector<uint64_t> &counts, const Config &config) { - std::size_t ret = Quant::Size(counts.size(), config) + Unigram::Size(counts[0]); + static uint64_t Size(const std::vector<uint64_t> &counts, const Config &config) { + uint64_t ret = Quant::Size(counts.size(), config) + Unigram::Size(counts[0]); for (unsigned char i = 1; i < counts.size() - 1; ++i) { ret += Middle::Size(Quant::MiddleBits(config), counts[i], counts[0], counts[i+1], config); } |