Method __init__

PATH/core/models/backbones/vit.py:32–34

(self, drop_prob=None)

    """

    def __init__(self, drop_prob=None):
        super(DropPath, self).__init__()
        self.drop_prob = drop_prob

    def forward(self, x):
        return drop_path(x, self.drop_prob, self.training)