spandrel_extra_arches.architectures.Restormer.Restormer

class documentation

class Restormer(nn.Module): (source)

Constructor: Restormer(inp_channels, out_channels, dim, num_blocks, ...)

Undocumented

Method	`__init__`	Undocumented
Method	`forward`	Undocumented
Class Variable	`hyperparameters`	Undocumented
Instance Variable	`decoder_level1`	Undocumented
Instance Variable	`decoder_level2`	Undocumented
Instance Variable	`decoder_level3`	Undocumented
Instance Variable	`down1_2`	Undocumented
Instance Variable	`down2_3`	Undocumented
Instance Variable	`down3_4`	Undocumented
Instance Variable	`dual_pixel_task`	Undocumented
Instance Variable	`encoder_level1`	Undocumented
Instance Variable	`encoder_level2`	Undocumented
Instance Variable	`encoder_level3`	Undocumented
Instance Variable	`latent`	Undocumented
Instance Variable	`output`	Undocumented
Instance Variable	`patch_embed`	Undocumented
Instance Variable	`reduce_chan_level2`	Undocumented
Instance Variable	`reduce_chan_level3`	Undocumented
Instance Variable	`refinement`	Undocumented
Instance Variable	`skip_conv`	Undocumented
Instance Variable	`up2_1`	Undocumented
Instance Variable	`up3_2`	Undocumented
Instance Variable	`up4_3`	Undocumented

def __init__(self, *, inp_channels=3, out_channels=3, dim=48, num_blocks=[4, 6, 6, 8], num_refinement_blocks=4, heads=[1, 2, 4, 8], ffn_expansion_factor=2.66, bias=False, LayerNorm_type='WithBias', dual_pixel_task=False): (source) ¶

Undocumented

def forward(self, inp_img): (source) ¶

Undocumented

hyperparameters: dict = (source) ¶

Undocumented

decoder_level1 = (source) ¶

Undocumented

decoder_level2 = (source) ¶

Undocumented

decoder_level3 = (source) ¶

Undocumented

down1_2 = (source) ¶