opencd/models/decode_heads/__init__.py

from .bit_head import BITHead
from .changer import Changer
from .general_scd_head import GeneralSCDHead
from .identity_head import DSIdentityHead, IdentityHead
from .multi_head import MultiHeadDecoder
from .sta_head import STAHead
from .tiny_head import TinyHead

__all__ = ['BITHead', 'Changer', 'IdentityHead', 'DSIdentityHead', 'TinyHead',
           'STAHead', 'MultiHeadDecoder', 'GeneralSCDHead']