spandrel.architectures.KBNet.KBNet

class documentation

class KBNet_s(nn.Module): (source)

Constructor: KBNet_s(img_channel, width, middle_blk_num, enc_blk_nums, ...)

Undocumented

Method	`__init__`	Undocumented
Method	`check_image_size`	Undocumented
Method	`forward`	Undocumented
Class Variable	`hyperparameters`	Undocumented
Instance Variable	`decoders`	Undocumented
Instance Variable	`downs`	Undocumented
Instance Variable	`encoders`	Undocumented
Instance Variable	`ending`	Undocumented
Instance Variable	`intro`	Undocumented
Instance Variable	`middle_blks`	Undocumented
Instance Variable	`padder_size`	Undocumented
Instance Variable	`ups`	Undocumented

def __init__(self, *, img_channel=3, width=64, middle_blk_num=12, enc_blk_nums=[2, 2, 4, 8], dec_blk_nums=[2, 2, 2, 2], lightweight=False, ffn_scale=2): (source) ¶

Undocumented

def check_image_size(self, x): (source) ¶

Undocumented

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

Undocumented

hyperparameters: dict = (source) ¶

Undocumented

decoders = (source) ¶

Undocumented

downs = (source) ¶

Undocumented

encoders = (source) ¶