Skip to content

GitLab

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

Standardize IPA formatting

parent 8d1119df
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
openfold/model/structure_module.py
View file @ 34e4e6ce
...@@ -291,7 +291,6 @@ class InvariantPointAttention(nn.Module): ...@@ -291,7 +291,6 @@ class InvariantPointAttention(nn.Module):
########################## ##########################
# Compute attention scores # Compute attention scores
########################## ##########################
# [*, N_res, N_res, H] # [*, N_res, N_res, H]
b = self.linear_b(z) b = self.linear_b(z)
...@@ -332,7 +331,6 @@ class InvariantPointAttention(nn.Module): ...@@ -332,7 +331,6 @@ class InvariantPointAttention(nn.Module):
################ ################
# Compute output # Compute output
################ ################
# [*, N_res, H, C_hidden] # [*, N_res, H, C_hidden]
o = torch.matmul(a, v.transpose(-2, -3)).transpose(-2, -3) 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