Method __init__

models/layers.py:54–58

(self, thresh=0.5, tau=0.25, gamma=1.0)

class LIFSpike(nn.Module):
    def __init__(self, thresh=0.5, tau=0.25, gamma=1.0):
        super(LIFSpike, self).__init__()
        self.thresh = thresh
        self.tau = tau
        self.gamma = gamma

    def forward(self, x):
        mem = torch.zeros_like(x[:, 0])