edit to lif

Author: ago109

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))