remove swin debug prints

rwightman · rwightman · commit 7b6a406474f2 · 2024-07-24T21:05:56.000-07:00
diff --git a/timm/models/swin_transformer_v2.py b/timm/models/swin_transformer_v2.py
@@ -333,14 +333,12 @@ def _calc_window_shift(
             target_shift_size = to_2tuple(target_shift_size)
 
         if self.always_partition:
-            print('ap', target_window_size, target_shift_size)
             return target_window_size, target_shift_size
 
         target_window_size = to_2tuple(target_window_size)
         target_shift_size = to_2tuple(target_shift_size)
         window_size = [r if r <= w else w for r, w in zip(self.input_resolution, target_window_size)]
         shift_size = [0 if r <= w else s for r, w, s in zip(self.input_resolution, window_size, target_shift_size)]
-        print('nap', window_size, shift_size)
         return tuple(window_size), tuple(shift_size)
 
     def set_input_size(
diff --git a/timm/models/swin_transformer_v2_cr.py b/timm/models/swin_transformer_v2_cr.py
@@ -289,8 +289,6 @@ def __init__(
             None if self.dynamic_mask else self.get_attn_mask(),
             persistent=False,
         )
-        print(self.dynamic_mask)
-
         self.init_weights()
 
     def _calc_window_shift(

Original file line number	Diff line number	Diff line change
`@@ -289,8 +289,6 @@ def __init__(`
`289`	`289`	`None if self.dynamic_mask else self.get_attn_mask(),`
`290`	`290`	`persistent=False,`
`291`	`291`	`)`
`292`		`- print(self.dynamic_mask)`
`293`		`-`
`294`	`292`	`self.init_weights()`
`295`	`293`
`296`	`294`	`def _calc_window_shift(`