9 lines
206 B
Python
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')
|