Update swiftformer.py

2023-07-26 01:32:27 +04:00
parent ef5daec20c
commit 37a4fe953d
1 changed files with 0 additions and 2 deletions
--- a/models/swiftformer.py
+++ b/models/swiftformer.py
@@ -209,7 +209,6 @@ class SwiftFormerLocalRepresentation(nn.Module):
                nn.init.constant_(m.bias, 0)

    def forward(self, x):
-        print("SwiftFormerLocalRepresentation input is ", x.shape)
        input = x
        x = self.dwconv(x)
        x = self.norm(x)
@@ -220,7 +219,6 @@ class SwiftFormerLocalRepresentation(nn.Module):
            x = input + self.drop_path(self.layer_scale * x)
        else:
            x = input + self.drop_path(x)
-        
        return x