From ee4d8fc69afc013769609c4ca5c83bf41f118cf0 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 5 Jul 2021 21:22:46 -0700 Subject: [PATCH] Remove unecessary line from nest post refactor --- timm/models/nest.py | 1 - 1 file changed, 1 deletion(-) diff --git a/timm/models/nest.py b/timm/models/nest.py index 601b6104..191d15c7 100644 --- 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)