edit to lif

NACLab · Jul 1, 2024 · f6091f1 · f6091f1
1 parent a45dbbd
commit f6091f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ngclearn/components/neurons/spiking/LIFCell.py b/ngclearn/components/neurons/spiking/LIFCell.py
@@ -237,8 +237,8 @@ def __init__(self, name, n_units, tau_m, resist_m=1., thr=-52., v_rest=-65.,
         self.n_units = n_units
 
         ## set up surrogate function for spike emission
-        #self.spike_fx, self.d_spike_fx = arctan_estimator() #
-        self.spike_fx, self.d_spike_fx = triangular_estimator() # straight_through_estimator()
+        self.spike_fx, self.d_spike_fx = arctan_estimator() #
+        #self.spike_fx, self.d_spike_fx = triangular_estimator() # straight_through_estimator()
 
         ## Compartment setup
         restVals = jnp.zeros((self.batch_size, self.n_units))