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])