add the SimpleBaseNode

fangwei123456 · Dec 4, 2023 · 6fbee6e · 6fbee6e
1 parent ec09d18
commit 6fbee6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spikingjelly/activation_based/neuron.py b/spikingjelly/activation_based/neuron.py
@@ -57,11 +57,11 @@ def neuronal_reset(self, spike):
 
         if self.v_reset is None:
             # soft reset
-            self.v = self.jit_soft_reset(self.v, spike_d, self.v_threshold)
+            self.v = self.v - self.v_threshold * spike_d
 
         else:
             # hard reset
-            self.v = self.jit_hard_reset(self.v, spike_d, self.v_reset)
+            self.v = spike_d * self.v_reset + (1. - spike_d) * self.v