mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Update metaformers.py
This commit is contained in:
parent
143f8e69b1
commit
dd57cde1bc
@ -453,7 +453,7 @@ class MetaFormerBlock(nn.Module):
|
||||
self.drop_path2(
|
||||
self.mlp(self.norm2(
|
||||
x.permute(0, 3, 1, 2)
|
||||
).permute(0, 2, 3, 1)
|
||||
)#.permute(0, 2, 3, 1)
|
||||
)
|
||||
)
|
||||
)
|
||||
|
Loading…
x
Reference in New Issue
Block a user