From a0f6b197cce4028a704b35c04a36bb889987adde Mon Sep 17 00:00:00 2001 From: ago109 Date: Sun, 7 Jul 2024 00:18:58 -0400 Subject: [PATCH] mod to stdp syn --- ngclearn/components/synapses/hebbian/STDPSynapse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ngclearn/components/synapses/hebbian/STDPSynapse.py b/ngclearn/components/synapses/hebbian/STDPSynapse.py index 3104cd7d0..393d95df7 100755 --- a/ngclearn/components/synapses/hebbian/STDPSynapse.py +++ b/ngclearn/components/synapses/hebbian/STDPSynapse.py @@ -95,7 +95,7 @@ def _compute_update(Aplus, Aminus, tau_plus, tau_minus, preSpike, postSpike, post_m = (post_tols > 0.) * 1. pre_m = (pre_tols > 0.) * 1. t_delta = ((weights * 0 + 1.) * post_tols) - pre_tols.T ## t_delta.shape = weights.shape - t_delta = t_delta * post_m * pre_m + t_delta = t_delta * post_m * pre_m.T ## calculate post-synaptic term postTerm = jnp.exp(-t_delta/tau_plus) dWpost = postTerm * (postSpike * Aplus)