From dcadaff8622e6c8b3450f12f45447f605c68556a Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 22 Jul 2026 03:33:44 -0500 Subject: method: condition apical vectorizers on neural state --- sdil/probes.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'sdil/probes.py') diff --git a/sdil/probes.py b/sdil/probes.py index 1698d75..0075bca 100644 --- a/sdil/probes.py +++ b/sdil/probes.py @@ -74,7 +74,7 @@ def alignment_report(net, x, y, y_onehot, cfg): negg = -g teaching, a, innovation = core.teaching_signal( net, l, c, h[l + 1], cfg, context) - Ac = c @ net.A[l].t() # pure vectorizer output + Ac = net.vectorizer(l, c, h[l + 1], context) # pure vectorizer output traffic = net.apical_traffic(l, h[l + 1], context) unexplained = traffic - net.baseline(l, h[l + 1]) # include the phi' gain that the plasticity rule actually applies, @@ -145,6 +145,8 @@ def _clone(net): n.W = [w.clone() for w in net.W] n.b = [b.clone() for b in net.b] n.A = [a.clone() for a in net.A] + n.A_gate = ([a.clone() for a in net.A_gate] + if getattr(net, "A_gate", None) is not None else None) n.P = [p.clone() for p in net.P] n.P_bias = ([p.clone() for p in net.P_bias] if getattr(net, "P_bias", None) is not None else None) -- cgit v1.2.3