A typo in Efficient Correlated layer

[hzhang57]

RAFT/core/corr.py

Line 82 in 25eb2ac

fmap1_i = self.pyramid[0][0].permute(0, 2, 3, 1).contiguous()

Hi, should the 0 here be i?

fmap1_i = self.pyramid[0][0].permute(0, 2, 3, 1).contiguous() --> fmap1_i = self.pyramid[i][0].permute(0, 2, 3, 1).contiguous()

[zachteed]

Hi, I believe the current implementation is correct. If you look the equation on page 7 of our paper, only the second feature my is pooled, fmap1 is not pooled

[hzhang57]

Thank you for detailed explanation, I miss understand the function.

[vadimkantorov]

@zachteed Then probably self.pyramid[i][0] should not even be computed for i > 0?