Update src/mlpack/methods/ann/layer/multihead_attention.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
2023-11-13 15:30:25 -05:00
parent d00446231a
commit dfcf40a71f
1 changed files with 2 additions and 1 deletions
@@ -211,7 +211,8 @@ class MultiheadAttentionType : public Layer<MatType>
      }
      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;
      }
    }