diff --git a/ngclearn/components/neurons/spiking/LIFCell.py b/ngclearn/components/neurons/spiking/LIFCell.py index 49dabe07..d208622f 100644 --- 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))