diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp index 328f81bb2e..95dc12cdad 100644 --- a/src/mlpack/methods/ann/layer/multihead_attention.hpp +++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp @@ -211,7 +211,8 @@ class MultiheadAttentionType : public Layer } if (srcSeqLen != tgtSeqLen) { - Log::Fatal << "If using selfAttention, srcSeqLen must equal tgtSeqLen" + Log::Fatal << "If using selfAttention, srcSeqLen (" << srcSeqLen + << " must equal tgtSeqLen " << tgtSeqLen << ")!" << std::endl; << std::endl; } }