From dfcf40a71f3cdf4ee80fc29ecab48d2a72c8fcd8 Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Mon, 13 Nov 2023 15:30:25 -0500 Subject: [PATCH] Update src/mlpack/methods/ann/layer/multihead_attention.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/ann/layer/multihead_attention.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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; } }