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