tweak

NACLab · Jul 1, 2024 · 0bf8bb3 · 0bf8bb3
1 parent f6091f1
commit 0bf8bb3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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])