modules/models/amt_splat.py

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


@register('amt_splat')
class AMT(BaseModel):
    def __init__(self, cfg):
        super(AMT, self).__init__(cfg)