(pretrained=False, in_22k=False, **kwargs)
| 175 | |
| 176 | @register_model |
| 177 | def convnext_base(pretrained=False, in_22k=False, **kwargs): |
| 178 | model = ConvNeXt(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) |
| 179 | if pretrained: |
| 180 | url = model_urls['convnext_base_22k'] if in_22k else model_urls['convnext_base_1k'] |
| 181 | checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu") |
| 182 | model.load_state_dict(checkpoint["model"]) |
| 183 | return model |
| 184 | |
| 185 | @register_model |
| 186 | def convnext_large(pretrained=False, in_22k=False, **kwargs): |
no test coverage detected