Method __init__

ML/src/python/neuralforge/nn/modules.py:107–109

(self, drop_prob=0.0)

class DropPath(nn.Module):
    def __init__(self, drop_prob=0.0):
        super().__init__()
        self.drop_prob = drop_prob
    
    def forward(self, x):
        if self.drop_prob == 0.0 or not self.training: