From 94bb5ff52191013bd4eaf3c144420c3bb0c276da Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Mon, 13 Nov 2023 15:45:43 -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 | 1 - 1 file changed, 1 deletion(-) diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp index 39072dbc46..a7d54a3ca8 100644 --- a/src/mlpack/methods/ann/layer/multihead_attention.hpp +++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp @@ -208,7 +208,6 @@ class MultiheadAttentionType : public Layer for (size_t i=2; iinputDimensions.size(); i++) { srcSeqLen *= this->inputDimensions[i]; - } if (srcSeqLen != tgtSeqLen) { Log::Fatal << "If using selfAttention, srcSeqLen (" << srcSeqLen