2020-07-07 19:32:06 +08:00

5 lines
109 B
Python

from .cls_head import ClsHead
from .linear_head import LinearClsHead
__all__ = ['ClsHead', 'LinearClsHead']