diff --git a/torchvision/prototype/models/depth/stereo/crestereo.py b/torchvision/prototype/models/depth/stereo/crestereo.py index 29c0be93618..bf5d7534c02 100644 --- a/torchvision/prototype/models/depth/stereo/crestereo.py +++ b/torchvision/prototype/models/depth/stereo/crestereo.py @@ -542,7 +542,7 @@ def forward( # 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)