Source code for mmpretrain.models.builder
# Copyright (c) OpenMMLab. All rights reserved.
from mmpretrain.registry import MODELS
BACKBONES = MODELS
NECKS = MODELS
HEADS = MODELS
LOSSES = MODELS
CLASSIFIERS = MODELS
RETRIEVER = MODELS
def build_retriever(cfg):
"""Build retriever."""
return RETRIEVER.build(cfg)