spandrel.architectures.SCUNet.SCUNet

class documentation

class SCUNet(nn.Module): (source)

Constructor: SCUNet(in_nc, config, dim, drop_path_rate, input_resolution)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method check_image_size Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable config Undocumented
Instance Variable dim Undocumented
Instance Variable head_dim Undocumented
Instance Variable m_body Undocumented
Instance Variable m_down1 Undocumented
Instance Variable m_down2 Undocumented
Instance Variable m_down3 Undocumented
Instance Variable m_head Undocumented
Instance Variable m_tail Undocumented
Instance Variable m_up1 Undocumented
Instance Variable m_up2 Undocumented
Instance Variable m_up3 Undocumented
Instance Variable window_size Undocumented
Method _init_weights Undocumented
def __init__(self, *, in_nc=3, config=[4, 4, 4, 4, 4, 4, 4], dim=64, drop_path_rate=0.0, input_resolution=256): (source)

Undocumented

def check_image_size(self, x): (source)

Undocumented

def forward(self, x0): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

config = (source)

Undocumented

dim = (source)

Undocumented

head_dim: int = (source)

Undocumented

m_body = (source)

Undocumented

m_down1 = (source)

Undocumented

m_down2 = (source)

Undocumented

m_down3 = (source)

Undocumented

m_head = (source)

Undocumented

m_tail = (source)

Undocumented

m_up1 = (source)

Undocumented

m_up2 = (source)

Undocumented

m_up3 = (source)

Undocumented

window_size: int = (source)

Undocumented

def _init_weights(self, m): (source)

Undocumented