spandrel.util.timm.DropBlock2d

class documentation

class DropBlock2d(nn.Module): (source)

Constructor: DropBlock2d(drop_prob, block_size, gamma_scale, with_noise, ...)

View In Hierarchy

DropBlock. See https://arxiv.org/pdf/1810.12890.pdf

Method __init__ Undocumented
Method forward Undocumented
Instance Variable batchwise Undocumented
Instance Variable block_size Undocumented
Instance Variable drop_prob Undocumented
Instance Variable fast Undocumented
Instance Variable gamma_scale Undocumented
Instance Variable inplace Undocumented
Instance Variable with_noise Undocumented
def __init__(self, drop_prob: float = 0.1, block_size: int = 7, gamma_scale: float = 1.0, with_noise: bool = False, inplace: bool = False, batchwise: bool = False, fast: bool = True): (source)

Undocumented

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

Undocumented

batchwise = (source)

Undocumented

block_size = (source)

Undocumented

drop_prob = (source)

Undocumented

fast = (source)

Undocumented

gamma_scale = (source)

Undocumented

inplace = (source)

Undocumented

with_noise = (source)

Undocumented