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

Co-authored-by: Ryan Curtin <ryan@ratml.org>
2023-11-13 15:45:43 -05:00
parent 4cb0cd111c
commit 94bb5ff521
1 changed files with 0 additions and 1 deletions
@@ -208,7 +208,6 @@ class MultiheadAttentionType : public Layer<MatType>
      for (size_t i=2; i<this->inputDimensions.size(); i++)
      {
        srcSeqLen *= this->inputDimensions[i];
-      }
      if (srcSeqLen != tgtSeqLen)
      {
        Log::Fatal << "If using selfAttention, srcSeqLen (" << srcSeqLen