spandrel.architectures.MoSR.MoSR

class documentation

class MoSR(nn.Module): (source)

Constructor: MoSR(in_ch, out_ch, upscale, n_block, ...)

View In Hierarchy

Mamba Out Super-Resolution

Method __init__ Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable gblocks Undocumented
Instance Variable shortcut Undocumented
Instance Variable upsampler Undocumented
def __init__(self, *, in_ch: int = 3, out_ch: int = 3, upscale: int = 4, n_block: int = 24, dim: int = 64, upsampler: str = 'ps', drop_path: float = 0.0, kernel_size: int = 7, expansion_ratio: float = 1.5, conv_ratio: float = 1.0): (source)

Undocumented

def forward(self, x): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

gblocks = (source)

Undocumented

shortcut = (source)

Undocumented

upsampler = (source)

Undocumented