Merge branch 'metaformer_baselines_for_vision' of https://github.com/fffffgggg54/pytorch-image-models into metaformer_baselines_for_vision

pull/1647/head
Fredo Guan 2 years ago
commit 5fe33b33a0

@ -615,7 +615,6 @@ class MetaFormerBlock(nn.Module):
def forward(self, x):
B, C, H, W = x.shape
x = x.view(B, H, W, C)
print(x.shape)
x = self.res_scale1(x) + \
self.layer_scale1(
self.drop_path1(

Loading…
Cancel
Save