From 37a4fe953d943c69ed844b6181bd1be59984fe49 Mon Sep 17 00:00:00 2001 From: Abdelrahman Shaker <108531886+Amshaker@users.noreply.github.com> Date: Wed, 26 Jul 2023 01:32:27 +0400 Subject: [PATCH] Update swiftformer.py --- models/swiftformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models/swiftformer.py b/models/swiftformer.py index 40597ae..1b74936 100644 --- 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