From 65ff44df46babcd110fadbbc452b6fdadeddb461 Mon Sep 17 00:00:00 2001 From: Monkin Date: Fri, 31 Jan 2025 17:12:44 +0000 Subject: [PATCH] Fix: Forward pass logic on snn.synaptic neuron --- snntorch/_neurons/synaptic.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/snntorch/_neurons/synaptic.py b/snntorch/_neurons/synaptic.py index fb30153f..d6a3c2db 100644 --- a/snntorch/_neurons/synaptic.py +++ b/snntorch/_neurons/synaptic.py @@ -214,9 +214,8 @@ def init_synaptic(self): return self.reset_mem() def forward(self, input_, syn=None, mem=None): - if not syn == None: - self.syn = mem + self.syn = syn if not mem == None: self.mem = mem @@ -252,9 +251,9 @@ def forward(self, input_, syn=None, mem=None): spk / self.graded_spikes_factor - self.reset ) # avoid double reset if self.reset_mechanism_val == 0: # reset by subtraction - mem = mem - do_reset * self.threshold + self.mem = self.mem - do_reset * self.threshold elif self.reset_mechanism_val == 1: # reset to zero - mem = mem - do_reset * mem + self.mem = self.mem - do_reset * self.mem if self.output: return spk, self.syn, self.mem