(self, drop_prob=0.0, scale_by_keep=True)
| 88 | class DropPath(nn.Module): |
| 89 | # adapted from https://github.com/huggingface/pytorch-image-models/blob/main/timm/layers/drop.py |
| 90 | def __init__(self, drop_prob=0.0, scale_by_keep=True): |
| 91 | super(DropPath, self).__init__() |
| 92 | self.drop_prob = drop_prob |
| 93 | self.scale_by_keep = scale_by_keep |
| 94 | |
| 95 | def forward(self, x): |
| 96 | if self.drop_prob == 0.0 or not self.training: |