diff options
| author | Hieu Hoang <hieuhoang@gmail.com> | 2017-06-20 17:27:41 +0300 |
|---|---|---|
| committer | Hieu Hoang <hieuhoang@gmail.com> | 2017-06-20 17:27:41 +0300 |
| commit | 1604516232133ffcc5d7cc47f2c9cb48fe771609 (patch) | |
| tree | 8a171ccb86a60f16f9761a1a66159e7944e9ac3b /src/amun/cpu/decoder/best_hyps.h | |
| parent | 4a58c88d3373583f15664a69b94f809a4b1e77e4 (diff) | |
| parent | b357f082f3d550197bef448ae71f6df02bbbd2bf (diff) | |
merge
Diffstat (limited to 'src/amun/cpu/decoder/best_hyps.h')
| -rw-r--r-- | src/amun/cpu/decoder/best_hyps.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/amun/cpu/decoder/best_hyps.h b/src/amun/cpu/decoder/best_hyps.h index 82d0f3e1..a148522f 100644 --- a/src/amun/cpu/decoder/best_hyps.h +++ b/src/amun/cpu/decoder/best_hyps.h @@ -7,6 +7,7 @@ #include "common/god.h" #include "common/exception.h" #include "cpu/mblas/matrix.h" +#include "cpu/decoder/encoder_decoder.h" namespace amunmt { namespace CPU { @@ -108,7 +109,7 @@ class BestHyps : public BestHypsBase if (returnAttentionWeights_) { std::vector<SoftAlignmentPtr> alignments; for (auto& scorer : scorers) { - if (CPU::EncoderDecoder* encdec = dynamic_cast<CPU::EncoderDecoder*>(scorer.get())) { + if (CPU::CPUEncoderDecoderBase* encdec = dynamic_cast<CPU::CPUEncoderDecoderBase*>(scorer.get())) { auto& attention = encdec->GetAttention(); alignments.emplace_back(new SoftAlignment(attention.begin(hypIndex), attention.end(hypIndex))); |