mmselfsup/openselfsup/models/registry.py

9 lines
206 B
Python

from openselfsup.utils import Registry
MODELS = Registry('model')
BACKBONES = Registry('backbone')
NECKS = Registry('neck')
HEADS = Registry('head')
MEMORIES = Registry('memory')
LOSSES = Registry('loss')