1ee6d86439ca85e5c8d6bf5f8112668b8b63de77,bindsnet/conversion/conversion.py,SubtractiveResetIFNodes,__init__,#SubtractiveResetIFNodes#,89

Before Change

            self.thresh = thresh

        // Post-spike refractory period.
        if isinstance(refrac, float):
            self.refrac = torch.tensor(refrac)
        else:
            self.refrac = refrac

        self.v = self.reset * torch.ones(self.shape)  // Neuron voltages.
        self.refrac_count = torch.zeros(self.shape)  // Refractory period counters.

    def forward(self, x: torch.Tensor) -> None:

After Change

        )

        self.register_buffer(
            "reset", torch.tensor(reset, dtype=torch.float)
        )  // Post-spike reset voltage.
        self.register_buffer(
            "thresh", torch.tensor(thresh, dtype=torch.float)
        )  // Spike threshold voltage.

Instances