diff --git a/timm/models/fasternet.py b/timm/models/fasternet.py index 67702b15..56225c4d 100644 --- a/timm/models/fasternet.py +++ b/timm/models/fasternet.py @@ -102,6 +102,7 @@ class MLPBlock(nn.Module): x = shortcut + self.drop_path(self.mlp(x)) return x + class Block(nn.Module): def __init__( self,