From 0bf8bb3be080d62cb6410cff2c259943a4302151 Mon Sep 17 00:00:00 2001 From: ago109 Date: Mon, 1 Jul 2024 15:29:50 -0400 Subject: [PATCH] tweak --- ngclearn/components/synapses/hebbian/hebbianSynapse.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ngclearn/components/synapses/hebbian/hebbianSynapse.py b/ngclearn/components/synapses/hebbian/hebbianSynapse.py index dbc3d544..7af5f45c 100644 --- a/ngclearn/components/synapses/hebbian/hebbianSynapse.py +++ b/ngclearn/components/synapses/hebbian/hebbianSynapse.py @@ -41,7 +41,8 @@ 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 - W * w_decay + dW = dW - jnp.matmul((1. - _pre).T, post) * w_decay + #dW = dW - W * w_decay return dW * signVal, db * signVal @partial(jit, static_argnums=[1,2])