diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp index e9b407e322..7466517e67 100644 --- a/src/mlpack/methods/ann/layer/multihead_attention.hpp +++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp @@ -192,7 +192,8 @@ class MultiheadAttentionType : public Layer void ComputeOutputDimensions() override { - if (this->inputDimensions.size() < 2) { + if (this->inputDimensions.size() < 2) + { Log::Fatal << "Must have at least two dimensions for MultiHeadAttention: " << " [EmbeddingDim,SequenceLen]" << std::endl; }