"git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "b61049fc6a7ec27442317bd04efc8e306d5e14b6"
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): ...@@ -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