From e6a43613063670439b01d5a6300f799c5da9261c Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 28 Aug 2022 15:27:01 -0700 Subject: [PATCH] pretrained_cfg entry for mvitv2_small_cls --- timm/models/mvitv2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/timm/models/mvitv2.py b/timm/models/mvitv2.py index 76e70135..c5aaa09e 100644 --- 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=''), )