Add update_feat_shape to dinov3 rope. Toggle fix_input_size bool to True on dinov3 pretrained_cfg so correct default size set if dynamic_img_size turned off. Fix #2593

Author: rwightman

timm/layers/pos_embed_sincos.py

@@ -1112,6 +1112,11 @@ def _cache_embed(self, feat_shape: List[int]):
         self.register_buffer("pos_embed_cached", rope_embed, persistent=False)
         self.feat_shape = feat_shape
 
+    def update_feat_shape(self, feat_shape: List[int]):
+        if self.feat_shape is not None and feat_shape != self.feat_shape:
+            # only update if feat_shape was set (valid cache) and different from previous value
+            self._cache_embed(feat_shape)
+
     def get_embed(self, shape: Optional[List[int]] = None) -> torch.Tensor:
         """Generate rope_embed matching DINOv3 RopePositionEmbedding numerics.
 

timm/models/eva.py

@@ -1302,7 +1302,7 @@ def _dinov3_cfg(url: str = '', **kwargs) -> Dict[str, Any]:
     return {
         'url': url,
         'num_classes': 0, 'input_size': (3, 256, 256), 'pool_size': None,
-        'crop_pct': 1.0, 'interpolation': 'bicubic', 'min_input_size': (3, 128, 128),
+        'crop_pct': 1.0, 'interpolation': 'bicubic', 'fixed_input_size': True,
         'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
         'first_conv': 'patch_embed.proj', 'classifier': 'head',
         'license': 'dinov3', **kwargs