From 9a5b6b13efa74efd9d9e2ca4c55d38879e5dad7a Mon Sep 17 00:00:00 2001 From: Josh Veitch-Michaelis Date: Thu, 10 Nov 2022 23:40:01 +0000 Subject: [PATCH] fix userwarning for rounded division --- mask2former/modeling/transformer_decoder/position_encoding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mask2former/modeling/transformer_decoder/position_encoding.py b/mask2former/modeling/transformer_decoder/position_encoding.py index f32532e0..336dda4b 100644 --- a/mask2former/modeling/transformer_decoder/position_encoding.py +++ b/mask2former/modeling/transformer_decoder/position_encoding.py @@ -38,7 +38,7 @@ def forward(self, x, mask=None): x_embed = x_embed / (x_embed[:, :, -1:] + eps) * self.scale dim_t = torch.arange(self.num_pos_feats, dtype=torch.float32, device=x.device) - dim_t = self.temperature ** (2 * (dim_t // 2) / self.num_pos_feats) + dim_t = self.temperature ** (2 * torch.div(dim_t, 2, rounding_mode='trunc') / self.num_pos_feats) pos_x = x_embed[:, :, :, None] / dim_t pos_y = y_embed[:, :, :, None] / dim_t