From b1c2e3eb92c85e460ca90133b93bbbf5476f927e Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 30 Sep 2021 23:19:05 -0700 Subject: [PATCH] Match rel_pos_indices attr rename in conv branch --- timm/models/layers/lambda_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/layers/lambda_layer.py b/timm/models/layers/lambda_layer.py index fd174855..eeb77e45 100644 --- 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