(pretrained=False, **kwargs)
| 462 | |
| 463 | @register_model |
| 464 | def pit_s_224(pretrained=False, **kwargs) -> PoolingVisionTransformer: |
| 465 | model_args = dict( |
| 466 | patch_size=16, |
| 467 | stride=8, |
| 468 | base_dims=[48, 48, 48], |
| 469 | depth=[2, 6, 4], |
| 470 | heads=[3, 6, 12], |
| 471 | mlp_ratio=4, |
| 472 | ) |
| 473 | return _create_pit('pit_s_224', pretrained, **dict(model_args, **kwargs)) |
| 474 | |
| 475 | |
| 476 | @register_model |
nothing calls this directly
no test coverage detected