2020-06-16 00:05:18 +08:00

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')