From 8ece2be7f9fce4deec40111c5f378b333bee8c52 Mon Sep 17 00:00:00 2001 From: ago109 Date: Mon, 1 Jul 2024 15:34:43 -0400 Subject: [PATCH] tweak --- ngclearn/components/synapses/hebbian/hebbianSynapse.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ngclearn/components/synapses/hebbian/hebbianSynapse.py b/ngclearn/components/synapses/hebbian/hebbianSynapse.py index 7af5f45c9..dbc3d5445 100644 --- a/ngclearn/components/synapses/hebbian/hebbianSynapse.py +++ b/ngclearn/components/synapses/hebbian/hebbianSynapse.py @@ -41,8 +41,7 @@ def _calc_update(pre, post, W, w_bound, is_nonnegative=True, signVal=1., w_decay if w_bound > 0.: dW = dW * (w_bound - jnp.abs(W)) if w_decay > 0.: - dW = dW - jnp.matmul((1. - _pre).T, post) * w_decay - #dW = dW - W * w_decay + dW = dW - W * w_decay return dW * signVal, db * signVal @partial(jit, static_argnums=[1,2])