Match rel_pos_indices attr rename in conv branch

3 years ago · b1c2e3eb92
parent b49630a138
commit b1c2e3eb92
1 changed files with 1 additions and 1 deletions
--- a/timm/models/layers/lambda_layer.py
+++ b/timm/models/layers/lambda_layer.py
@ -67,7 +67,7 @@ class LambdaLayer(nn.Module):
            # local lambda convolution for pos
            self.conv_lambda = nn.Conv3d(1, dim_head, (r, r, 1), padding=(r // 2, r // 2, 0))
            self.pos_emb = None
-            self.rel_pos = None
+            self.rel_pos_indices = None
        else:
            # relative pos embedding
            assert feat_size is not None