apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Adam Kropp
2023-11-13 13:47:17 -05:00
committed by GitHub
co-authored by Ryan Curtin
parent f5be365000
commit eb3e7f2a48
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/multihead_attention.hpp
+2 -1
View File
@@ -192,7 +192,8 @@ class MultiheadAttentionType : public Layer<MatType>
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;
}