spandrel.architectures.KBNet.KBNet_l

class documentation

class KBNet_l(nn.Module): (source)

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

View In Hierarchy

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 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 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=1.5, 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)

Undocumented

down2_3 = (source)

Undocumented

down3_4 = (source)

Undocumented

encoder_level1 = (source)

Undocumented

encoder_level2 = (source)

Undocumented

encoder_level3 = (source)

Undocumented

latent = (source)

Undocumented

output = (source)

Undocumented

patch_embed = (source)

Undocumented

reduce_chan_level2 = (source)

Undocumented

reduce_chan_level3 = (source)

Undocumented

refinement = (source)

Undocumented

up2_1 = (source)

Undocumented

up3_2 = (source)

Undocumented

up4_3 = (source)

Undocumented