Skip to content

Commit eb94efb

Browse files
wojtkerwightman
authored andcommitted
fix pos embed dynamic resampling for eva
1 parent 3c7822c commit eb94efb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Diff for: timm/models/eva.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -560,9 +560,11 @@ def _pos_embed(self, x) -> Tuple[torch.Tensor, Optional[torch.Tensor]]:
560560
if self.dynamic_img_size:
561561
B, H, W, C = x.shape
562562
if self.pos_embed is not None:
563+
prev_grid_size = self.patch_embed.grid_size
563564
pos_embed = resample_abs_pos_embed(
564565
self.pos_embed,
565-
(H, W),
566+
new_size=(H, W),
567+
old_size=prev_grid_size,
566568
num_prefix_tokens=self.num_prefix_tokens,
567569
)
568570
else:

0 commit comments

Comments
 (0)