Commit 34e4e6ce authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Standardize IPA formatting

parent 8d1119df
......@@ -291,7 +291,6 @@ class InvariantPointAttention(nn.Module):
##########################
# Compute attention scores
##########################
# [*, N_res, N_res, H]
b = self.linear_b(z)
......@@ -332,7 +331,6 @@ class InvariantPointAttention(nn.Module):
################
# Compute output
################
# [*, N_res, H, C_hidden]
o = torch.matmul(a, v.transpose(-2, -3)).transpose(-2, -3)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment