(variant, pretrained=False, default_cfg=None, fuse=False, **kwargs)
| 548 | |
| 549 | |
| 550 | def create_levit(variant, pretrained=False, default_cfg=None, fuse=False, **kwargs): |
| 551 | if kwargs.get('features_only', None): |
| 552 | raise RuntimeError('features_only not implemented for Vision Transformer models.') |
| 553 | |
| 554 | model_cfg = dict(**model_cfgs[variant], **kwargs) |
| 555 | model = build_model_with_cfg( |
| 556 | Levit, variant, pretrained, |
| 557 | default_cfg=default_cfgs[variant], |
| 558 | pretrained_filter_fn=checkpoint_filter_fn, |
| 559 | **model_cfg) |
| 560 | #if fuse: |
| 561 | # utils.replace_batchnorm(model) |
| 562 | return model |
| 563 |
no test coverage detected