Class Config

tgs/utils/base.py:90–92

class BaseModule(nn.Module, Updateable):
    @dataclass
    class Config:
        weights: Optional[str] = None
        freeze: Optional[bool] = False

    cfg: Config  # add this to every subclass of BaseModule to enable static type checking