From 8c26f3829ac8ded5f105d48c68a521d1c5149369 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 14 Dec 2024 21:15:48 -0700 Subject: [PATCH] Update dependencyvit.py --- timm/models/dependencyvit.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py index 4fa8b7ca7..6734466f9 100644 --- a/timm/models/dependencyvit.py +++ b/timm/models/dependencyvit.py @@ -350,6 +350,13 @@ def dependencyvit_tiny_patch16_224(pretrained: bool = False, **kwargs) -> Depend model = _create_dependencyvit('dependencyvit_tiny_patch16_224', pretrained=pretrained, **dict(model_args, **kwargs)) return model +@register_model +def dependencyvit_tiny_cpe5_patch16_224(pretrained: bool = False, **kwargs) -> DependencyViT: + model_args = dict(patch_size=16, embed_dim=192, depth=12, num_heads=12, cpe_depth=5) + model = _create_dependencyvit('dependencyvit_tiny_cpe5_patch16_224', pretrained=pretrained, **dict(model_args, **kwargs)) + return model + + @register_model def dependencyvit_small_patch16_224(pretrained: bool = False, **kwargs) -> DependencyViT: model_args = dict(patch_size=16, embed_dim=384, depth=12, num_heads=12)