pretrained_cfg entry for mvitv2_small_cls

2022-08-28 15:27:01 -07:00 · 2022-08-28 15:27:01 -07:00 · e6a4361306
parent f66e5f0e35
commit e6a4361306
1 changed files with 2 additions and 0 deletions
--- a/timm/models/mvitv2.py
+++ b/timm/models/mvitv2.py
@ -57,6 +57,8 @@ default_cfgs = dict(
    mvitv2_huge_in21k=_cfg(
        url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_H_in21k.pyth',
        num_classes=19168),
+
+    mvitv2_small_cls=_cfg(url=''),
 )