(variant, cfg_variant=None, pretrained=False, distilled=True, **kwargs)
| 963 | |
| 964 | |
| 965 | def create_levit(variant, cfg_variant=None, pretrained=False, distilled=True, **kwargs): |
| 966 | is_conv = '_conv' in variant |
| 967 | out_indices = kwargs.pop('out_indices', (0, 1, 2)) |
| 968 | if kwargs.get('features_only', False) and not is_conv: |
| 969 | kwargs.setdefault('feature_cls', 'getter') |
| 970 | if cfg_variant is None: |
| 971 | if variant in model_cfgs: |
| 972 | cfg_variant = variant |
| 973 | elif is_conv: |
| 974 | cfg_variant = variant.replace('_conv', '') |
| 975 | |
| 976 | model_cfg = dict(model_cfgs[cfg_variant], **kwargs) |
| 977 | model = build_model_with_cfg( |
| 978 | LevitDistilled if distilled else Levit, |
| 979 | variant, |
| 980 | pretrained, |
| 981 | pretrained_filter_fn=checkpoint_filter_fn, |
| 982 | feature_cfg=dict(flatten_sequential=True, out_indices=out_indices), |
| 983 | **model_cfg, |
| 984 | ) |
| 985 | return model |
| 986 | |
| 987 | |
| 988 | def _cfg(url='', **kwargs): |
no test coverage detected