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 | a36fcc5d55c1de84ae68c1091ebff2b1c32dc3b7 (patch) | |
| tree | 1c22c35065e52c5ae69fa915bd502939be801374 /klm/lm/search_hashed.hh | |
| parent | 8a10fd2856e13489cc10d2c683ecbce687f1e5ff (diff) | |
KenLM 8307220
Diffstat (limited to 'klm/lm/search_hashed.hh')
| -rw-r--r-- | klm/lm/search_hashed.hh | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/klm/lm/search_hashed.hh b/klm/lm/search_hashed.hh index 7e8c1220..a52f107b 100644 --- a/klm/lm/search_hashed.hh +++ b/klm/lm/search_hashed.hh @@ -74,8 +74,8 @@ template <class Value> class HashedSearch { // TODO: move probing_multiplier here with next binary file format update. static void UpdateConfigFromBinary(int, const std::vector<uint64_t> &, Config &) {} - static std::size_t Size(const std::vector<uint64_t> &counts, const Config &config) { - std::size_t ret = Unigram::Size(counts[0]); + static uint64_t Size(const std::vector<uint64_t> &counts, const Config &config) { + uint64_t ret = Unigram::Size(counts[0]); for (unsigned char n = 1; n < counts.size() - 1; ++n) { ret += Middle::Size(counts[n], config.probing_multiplier); } @@ -160,8 +160,8 @@ template <class Value> class HashedSearch { #endif {} - static std::size_t Size(uint64_t count) { - return (count + 1) * sizeof(ProbBackoff); // +1 for hallucinate <unk> + static uint64_t Size(uint64_t count) { + return (count + 1) * sizeof(typename Value::Weights); // +1 for hallucinate <unk> } const typename Value::Weights &Lookup(WordIndex index) const { |