mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Remove unecessary line from nest post refactor

Browse Source
This commit is contained in:
Ross Wightman 2021-07-05 21:22:46 -07:00
parent c8ec1ffcb9
commit ee4d8fc69a
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/nest.py
View File

@ -329,7 +329,6 @@ class Nest(nn.Module):
def forward_features(self, x): def forward_features(self, x):
""" x shape (B, C, H, W) """ x shape (B, C, H, W)
""" """
B, _, H, W = x.shape
x = self.patch_embed(x) x = self.patch_embed(x)
x = self.levels(x) x = self.levels(x)
# Layer norm done over channel dim only (to NHWC and back) # Layer norm done over channel dim only (to NHWC and back)