spandrel.architectures.DITN.DITN

class documentation

class DITN(nn.Module): (source)

Constructor: DITN(inp_channels, dim, ITL_blocks, SAL_blocks, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method check_image_size Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable conv_after_body Undocumented
Instance Variable dim Undocumented
Instance Variable ITL_blocks Undocumented
Instance Variable patch_size Undocumented
Instance Variable patch_sizes Undocumented
Instance Variable SAL_blocks Undocumented
Instance Variable scale Undocumented
Instance Variable sft Undocumented
Instance Variable UFONE Undocumented
Instance Variable upsample Undocumented
def __init__(self, *, inp_channels=3, dim=60, ITL_blocks=4, SAL_blocks=4, UFONE_blocks=1, ffn_expansion_factor=2, bias=False, LayerNorm_type='WithBias', patch_size=8, upscale=4): (source)

Undocumented

def check_image_size(self, x): (source)

Undocumented

def forward(self, inp_img): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

conv_after_body = (source)

Undocumented

dim = (source)

Undocumented

ITL_blocks = (source)

Undocumented

patch_size = (source)

Undocumented

patch_sizes: list[int] = (source)

Undocumented

SAL_blocks = (source)

Undocumented

scale = (source)

Undocumented

sft = (source)

Undocumented

UFONE = (source)

Undocumented

upsample = (source)

Undocumented