Update davit.py

3 years ago · 2de09f6bb7
parent d8cce7d1e4
commit 2de09f6bb7
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -28,7 +28,7 @@ from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, Cl
 from .features import FeatureInfo
 from collections import OrderedDict
 import torch.utils.checkpoint as checkpoint
-from .pretrained import generate_default_cfgs
+from .pretrained import generate_default_cfgs, pretrained_cfg_for_features
 from .registry import register_model