from modules.models.base_model import BaseModel from modules.models import register @register('amt_bilateral') class AMT(BaseModel): def __init__(self, cfg): super(AMT, self).__init__(cfg)