Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Oct 9, 2023

Commit

7fe1144

1 Parent(s): 738b546

Fix Einstein summation notation

Files changed (1) hide show

qarac/models/layers/GlobalAttentionPoolingHead.py CHANGED Viewed

@@ -66,5 +66,5 @@ class GlobalAttentionPoolingHead(torch.nn.Module):
                                                       dim=2,
                                                       keepdim=True),EPSILON)
         lp = loc/nloc
-        attention = torch.einsum('ijk,k->ij',lp,gp)
         return torch.einsum('ij,ijk->ik',attention,Xa)

                                                       dim=2,
                                                       keepdim=True),EPSILON)
         lp = loc/nloc
+        attention = torch.einsum('ijk,ik->ij',lp,gp)
         return torch.einsum('ij,ijk->ik',attention,Xa)