diff --git a/nets/corr.py b/nets/corr.py index 446500a..15df9d2 100644 --- a/nets/corr.py +++ b/nets/corr.py @@ -36,7 +36,7 @@ def get_correlation(self, left_feature, right_feature, psize=(3, 3), dilate=(1, di_y, di_x = dilate[0], dilate[1] pady, padx = psize[0] // 2 * di_y, psize[1] // 2 * di_x - right_pad = F.pad(right_feature, pad_witdth=( + right_pad = F.pad(right_feature, pad_width=( (0, 0), (0, 0), (pady, pady), (padx, padx)), mode="replicate") right_slid = F.sliding_window(