mmselfsup/openselfsup/models/heads/__init__.py

4 lines
117 B
Python
Raw Normal View History

2020-06-16 00:05:18 +08:00
from .contrastive_head import ContrastiveHead
from .cls_head import ClsHead
from .multi_cls_head import MultiClsHead