spandrel.architectures.DRUNet.DRUNet

class documentation

class DRUNet(nn.Module): (source)

Constructor: DRUNet(in_nc, out_nc, nc, nb, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Class Variable hyperparameters 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
def __init__(self, *, in_nc=1, out_nc=1, nc=[64, 128, 256, 512], nb=4, act_mode='R', downsample_mode='strideconv', upsample_mode='convtranspose'): (source)

Undocumented

def forward(self, x0): (source)

Undocumented

hyperparameters: dict = (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