Remove unecessary line from nest post refactor

4 years ago · ee4d8fc69a
parent c8ec1ffcb9
commit ee4d8fc69a
1 changed files with 0 additions and 1 deletions
--- a/timm/models/nest.py
+++ b/timm/models/nest.py
@ -329,7 +329,6 @@ class Nest(nn.Module):
    def forward_features(self, x):
        """ x shape (B, C, H, W)
        """
-        B, _, H, W = x.shape
        x = self.patch_embed(x)
        x = self.levels(x)
        # Layer norm done over channel dim only (to NHWC and back)