diff --git a/timm/models/davit.py b/timm/models/davit.py index e0bb0aa0..18898200 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -384,7 +384,7 @@ class DaViT(nn.Module): num_classes=1000, global_pool='avg'#, #features_only = False - ): + **kwargs): super().__init__() architecture = [[index] * item for index, item in enumerate(depths)]