spandrel.architectures.NAFNet.NAFNet

class documentation

class NAFNet(nn.Module): (source)

Constructor: NAFNet(img_channel, width, middle_blk_num, enc_blk_nums, dec_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: int = 3, width: int = 16, middle_blk_num: int = 1, enc_blk_nums: list[int] = [], dec_blk_nums: list[int] = []): (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) ¶