mod to gated-hebb

NACLab · Jul 1, 2024 · 94b2453 · 94b2453
1 parent 429f855
commit 94b2453
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ngclearn/components/synapses/hebbian/gatedHebbianSynapse.py b/ngclearn/components/synapses/hebbian/gatedHebbianSynapse.py
@@ -43,11 +43,11 @@ def _evolve(bias_init, eta, w_decay, w_bound, pre, post, preSpike,
         ## calculate synaptic update values
         dWeights, dBiases = GatedHebbianSynapse._compute_update(w_bound, pre, post, weights)
         weights = weights + dWeights * eta
-        if bias_init != None:
-            biases = biases + dBiases * eta
         if w_decay > 0.:
             Wdec = jnp.matmul((1. - preSpike).T, postSpike) * w_decay
             weights = weights - Wdec
+        if bias_init != None:
+            biases = biases + dBiases * eta
         weights = jnp.clip(weights, 0., w_bound)
         return weights, biases, dWeights, dBiases