spandrel.util.timm.DropPath

class documentation

class DropPath(nn.Module): (source)

Constructor: DropPath(drop_prob, scale_by_keep)

View In Hierarchy

Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).

Method __init__ Undocumented
Method extra_repr Undocumented
Method forward Undocumented
Instance Variable drop_prob Undocumented
Instance Variable scale_by_keep Undocumented
def __init__(self, drop_prob: float = 0.0, scale_by_keep: bool = True): (source)

Undocumented

def extra_repr(self): (source)

Undocumented

def forward(self, x: torch.Tensor): (source)

Undocumented

drop_prob = (source)

Undocumented

scale_by_keep = (source)

Undocumented