Skip to content

GitLab

Unverified Commit 9d322cac authored by Joao Gomes's avatar Joao Gomes Committed by GitHub
Browse files

[FBcode->GH] fix typo in CRES stereo (#6813) 

[FBcode->GH]
parent 121a780c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/prototype/models/depth/stereo/crestereo.py
View file @ 9d322cac
...@@ -542,7 +542,7 @@ class LocalFeatureEncoderLayer(nn.Module): ...@@ -542,7 +542,7 @@ class LocalFeatureEncoderLayer(nn.Module):
# attention operation # attention operation
message = self.attention_op(queries, keys, values, x_mask, source_mask) message = self.attention_op(queries, keys, values, x_mask, source_mask)
# concatenating attention heads together before passing throught projection layer # concatenating attention heads together before passing through projection layer
message = self.merge(message.reshape(B, S, D)) message = self.merge(message.reshape(B, S, D))
message = self.attention_norm(message) message = self.attention_norm(message)
......
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