(self, drop_prob=None, scale_by_keep=True)
| 429 | """ |
| 430 | |
| 431 | def __init__(self, drop_prob=None, scale_by_keep=True): |
| 432 | super(DropPath, self).__init__() |
| 433 | self.drop_prob = drop_prob |
| 434 | self.scale_by_keep = scale_by_keep |
| 435 | |
| 436 | def forward(self, x): |
| 437 | return drop_path(x, self.drop_prob, self.training, self.scale_by_keep) |