minor mod to lif

NACLab · Jul 23, 2024 · 23eb7ed · 23eb7ed
1 parent 2189f51
commit 23eb7ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ngclearn/components/neurons/spiking/LIFCell.py b/ngclearn/components/neurons/spiking/LIFCell.py
@@ -277,7 +277,7 @@ def _advance_state(t, dt, tau_m, R_m, v_rest, v_reset, v_decay, refract_T,
             thr_theta = _update_theta(dt, thr_theta, raw_spikes, tau_theta, theta_plus)
         ## update tols
         tols = _update_times(t, s, tols)
-        return v, s, raw_spikes, rfr, thr_theta, tols, key, surrogate
+        return jnp.maximum(v, v_rest), s, raw_spikes, rfr, thr_theta, tols, key, surrogate
 
     @resolver(_advance_state)
     def advance_state(self, v, s, s_raw, rfr, thr_theta, tols, key, surrogate):