From 34529ed363a46c1f7843b546fe577051290da2c7 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 11:36:58 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index b3f02b0a..6079554d 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -215,8 +215,8 @@ def window_reverse(windows : Tensor, window_size: int, H: int, W: int): Returns: x: (B, H, W, C) """ - window_size_dim_0 : int = windows.size(dim=0) - B = int(window_size_dim_0 / (H * W / window_size / window_size)) + + B = torch.floor(torch.Tensor(windows.size(dim=0)) / (H * W / window_size / window_size)) x = windows.view(B, H // window_size, W // window_size, window_size, window_size, -1) x = x.permute(0, 1, 3, 2, 4, 5).contiguous().view(B, H, W, -1) return x