del print

qubvel-org · Dec 5, 2024 · 8d15c28 · 8d15c28
1 parent 1901eb7
commit 8d15c28
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/segmentation_models_pytorch/decoders/pan/decoder.py b/segmentation_models_pytorch/decoders/pan/decoder.py
@@ -123,9 +123,7 @@ def forward(self, x):
         mid = self.mid(x)
         x1 = self.down1(x)
         x2 = self.down2(x1)
-        print(x2.shape)
         x3 = self.down3(x2)
-        print(x3.shape)
         x3 = F.interpolate(x3, size=(h // 4, w // 4), **upscale_parameters)
 
         x2 = self.conv2(x2)