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

[FBcode->GH]

[FBcode->GH] fix typo in CRES stereo (#6813)
[FBcode->GH]
9d322cac · Joao Gomes · GitHub · 121a780c · 9d322cac
Unverified Commit 9d322cac authored Oct 21, 2022 by Joao Gomes Committed by GitHub Oct 21, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/prototype/models/depth/stereo/crestereo.py torchvision/prototype/models/depth/stereo/crestereo.py +1 -1

No files found.
--- a/torchvision/prototype/models/depth/stereo/crestereo.py
+++ b/torchvision/prototype/models/depth/stereo/crestereo.py
@@ -542,7 +542,7 @@ class LocalFeatureEncoderLayer(nn.Module):

        # attention operation
        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.attention_norm(message)