spandrel_extra_arches.architectures.M3SNet.M3SNet

class documentation

class M3SNet(nn.Module): (source)

Constructor: M3SNet(img_channel, width, middle_blk_num, enc_blk_nums, dec_blk_nums)

View In Hierarchy

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`	Undocumented
Instance Variable	`middle_blks`	Undocumented
Instance Variable	`middle_ups`	Undocumented
Instance Variable	`padder_size`	Undocumented
Instance Variable	`ups`	Undocumented

def __init__(self, *, img_channel=3, width=32, middle_blk_num=1, enc_blk_nums=[1, 1, 1, 28], dec_blk_nums=[1, 1, 1, 1]): (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) ¶

Undocumented

ending = (source) ¶

Undocumented

intro = (source) ¶

Undocumented

middle = (source) ¶

Undocumented

middle_blks = (source) ¶

Undocumented

middle_ups = (source) ¶

Undocumented

padder_size = (source) ¶

Undocumented

ups = (source) ¶

Undocumented