spandrel.architectures.FFTformer.FFTformer

class documentation

class FFTformer(nn.Module): (source)

Constructor: FFTformer(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	`encoder_level1`	Undocumented
Instance Variable	`encoder_level2`	Undocumented
Instance Variable	`encoder_level3`	Undocumented
Instance Variable	`fuse1`	Undocumented
Instance Variable	`fuse2`	Undocumented
Instance Variable	`output`	Undocumented
Instance Variable	`patch_embed`	Undocumented
Instance Variable	`reduce_chan_level2`	Undocumented
Instance Variable	`refinement`	Undocumented
Instance Variable	`up2_1`	Undocumented
Instance Variable	`up3_2`	Undocumented

def __init__(self, *, inp_channels=3, out_channels=3, dim=48, num_blocks=[6, 6, 12], num_refinement_blocks=4, ffn_expansion_factor=3, bias=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) ¶