From acda5b8ff136d4474189f4a20e245129fe9438cb Mon Sep 17 00:00:00 2001 From: Jacek Chmielewski Date: Wed, 14 Feb 2024 10:29:30 +0100 Subject: [PATCH] Fix F.interpolate size tuple typing. Fixes #246 by casting numpy.int64 to int. --- midas/backbones/beit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/midas/backbones/beit.py b/midas/backbones/beit.py index 7a24e02c..8f25dd1e 100644 --- a/midas/backbones/beit.py +++ b/midas/backbones/beit.py @@ -44,7 +44,7 @@ def _get_rel_pos_bias(self, window_size): old_sub_table = old_relative_position_bias_table[:old_num_relative_distance - 3] old_sub_table = old_sub_table.reshape(1, old_width, old_height, -1).permute(0, 3, 1, 2) - new_sub_table = F.interpolate(old_sub_table, size=(new_height, new_width), mode="bilinear") + new_sub_table = F.interpolate(old_sub_table, size=(int(new_height), int(new_width)), mode="bilinear") new_sub_table = new_sub_table.permute(0, 2, 3, 1).reshape(new_num_relative_distance - 3, -1) new_relative_position_bias_table = torch.cat(