spandrel.architectures.PLKSR.PLKSR

class documentation

class PLKSR(nn.Module): (source)

Constructor: PLKSR(dim, n_blocks, upscaling_factor, ccm_type, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable feats Undocumented
Instance Variable repeat_op Undocumented
Instance Variable to_img Undocumented
def __init__(self, *, dim: int = 64, n_blocks: int = 28, upscaling_factor: int = 4, ccm_type: Literal['CCM', 'ICCM', 'DCCM'] = 'CCM', kernel_size: int = 17, split_ratio: float = 0.25, lk_type: Literal['PLK', 'SparsePLK', 'RectSparsePLK'] = 'PLK', use_max_kernel: bool = False, sparse_kernels: Sequence[int] = [5, 5, 5, 5], sparse_dilations: Sequence[int] = [1, 2, 3, 4], with_idt: bool = False, use_ea: bool = True): (source)

Undocumented

def forward(self, x: torch.Tensor) -> torch.Tensor: (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

feats = (source)

Undocumented

repeat_op = (source)

Undocumented

to_img = (source)

Undocumented