(pretrained_cfg: Dict[str, Any])
| 289 | |
| 290 | |
| 291 | def pretrained_cfg_for_features(pretrained_cfg: Dict[str, Any]) -> Dict[str, Any]: |
| 292 | pretrained_cfg = deepcopy(pretrained_cfg) |
| 293 | # remove default pretrained cfg fields that don't have much relevance for feature backbone |
| 294 | to_remove = ('num_classes', 'classifier', 'global_pool') # add default final pool size? |
| 295 | for tr in to_remove: |
| 296 | pretrained_cfg.pop(tr, None) |
| 297 | return pretrained_cfg |
| 298 | |
| 299 | |
| 300 | def _filter_kwargs(kwargs: Dict[str, Any], names: List[str]) -> None: |
no outgoing calls
no test coverage detected