| 157 | |
| 158 | @register_model |
| 159 | def convnext_tiny(pretrained=False,in_22k=False, **kwargs): |
| 160 | model = ConvNeXt(depths=[3, 3, 9, 3], dims=[96, 192, 384, 768], **kwargs) |
| 161 | if pretrained: |
| 162 | url = model_urls['convnext_tiny_22k'] if in_22k else model_urls['convnext_tiny_1k'] |
| 163 | checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", check_hash=True) |
| 164 | model.load_state_dict(checkpoint["model"]) |
| 165 | return model |
| 166 | |
| 167 | @register_model |
| 168 | def convnext_small(pretrained=False,in_22k=False, **kwargs): |