modules/models/upr_basic.py

from modules.models.base_model import BaseModel
from modules.models import register


@register('upr_basic')
class UPRNet(BaseModel):
    def __init__(self, cfg):
        super(UPRNet, self).__init__(cfg)